尝试用Java编写代码...我创建了一个GraphNode类来表示具有指向其父级的指针的节点。
我还创建了一个DisjointSet类,其中包含一个MakeSet方法,该方法创建一个GraphNode对象并使其父引用引用自身。
问题是:我如何存储每个节点,以便稍后在Union和FindSet中轻松访问它?我首先想到它将它存储在BST中,但是我必须创建一个自定义的TreeNode类,它不仅存储值,还存储对GraphNode的引用。有更简单的方法吗?
答案 0 :(得分:8)
绝对有一种更简单的方法:忘记所有节点业务。节点只是概念性的,不需要按字面意思实现它们,并且更容易不这样做。
您需要的只是两个整数数组。存储父母的人和存储队伍的人。所以在一种伪代码中,它看起来像这样:
disjoint_set {
int[] parent, rank;
makeset(int n)
{
rank = new int[n];
parent = new int[n];
for(int i = 0; i < n; i++)
parent[i] = i;
}
int find(int i)
{
if (parent[i] != i)
parent[i] = find(parent[i]);
return parent[i];
}
void union(int x, int y)
{
x_root = find(x);
y_root = find(y);
if (x_root != y_root) {
if (rank[x_root] < rank[y_root])
parent[x_root] = y_root;
else if (rank[x_root] > rank[y_root])
parent[y_root] = x_root;
else {
parent[y_root] = x_root;
rank[x_root]++;
}
}
}
}