如何在一个不相交的森林中存储每一组?

时间:2014-02-24 23:56:00

标签: java data-structures disjoint-sets

尝试用Java编写代码...我创建了一个GraphNode类来表示具有指向其父级的指针的节点。

我还创建了一个DisjointSet类,其中包含一个MakeSet方法,该方法创建一个GraphNode对象并使其父引用引用自身。

问题是:我如何存储每个节点,以便稍后在Union和FindSet中轻松访问它?我首先想到它将它存储在BST中,但是我必须创建一个自定义的TreeNode类,它不仅存储值,还存储对GraphNode的引用。有更简单的方法吗?

1 个答案:

答案 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]++;
            }
        }
    }
}