Question

我正在尝试为Kruskal实现union-find alg。我正在使用这个伪代码，我不理解下面的union部分step2（它不是递归调用）或者我是否接近。如果这种方式不起作用，我可以使用任何实现，只要我理解它。提前谢谢。 U和V是我的边缘节点，现在只是整数。

  Init(V)
  1.for every vertex v do
  2.boss[v]=v
  3.size[v]=1
  4.set[v]={v}

  Find (u)
  1.Return boss[u] 

  Union (u,v)
  1.if size[boss[u]]>size[boss[v]] then
  2.set[boss[u]]=set[boss[u]] union set[boss[v]]
  3.size[boss[u]]+=size[boss[v]]
  4.for every z in set[boss[v]] do
  5.boss[z]=boss[u]
  6.else do steps 2.-5. with u,v switched

我不明白第2步，到目前为止这是我的代码：

public class UnionFind {

    private int[] _boss;
    private int[] _size;
    private int[] _set;

    public UnionFind(int max) 
    {
        _boss = new int[max];
        _size = new int[max];
        _set  = new int[max];
    }

    public void init(Set<Integer> vertSet)
    {
        //for every vertex do
        int j=0;
        for(int i : vertSet)
        {
            _boss[j]=i;
            _size[j]=1;
            _set[j]=i;
            j++;
        }
    }

    int find(int u)
    {
        return(_boss[u]);
    }

    void union(int u, int v)
    {
        if(_size[_boss[u]]>_size[_boss[v]])
        {
            _set[_boss[u]]=_set[_boss[u]];
             //_set[_boss[v]];
            _size[_boss[u]]+=_size[_boss[v]];

            for(int z=0;z<_boss.length;z++)
            {
                _boss[z]=_boss[u];
            }
        }
        else
        {
            //switch u and v
            _set[_boss[v]]=_set[_boss[v]];
            //union(_set[_boss[v]],_set[_boss[u]]);
            _size[_boss[v]]+=_size[_boss[u]];

            for(int z=0;z<_boss.length;z++)
            {
                _boss[z]=_boss[v];
            }
        }
    }

Answer 1

步骤2意味着集合u现在成为u和v的并集，所以你不能分配set [v] = set [v]（你和v是boss [u]，boss [v]的缩写，分别）。因此，如果设置u是{u}并且设置v是{v}，则在此步骤之后，设置u将是{u，v}，再设置一个示例u = {1,2}，设置v = {3， 4}，所以在此之后，设置u = {1,2,3,4}。根据您使用的数据结构，您需要以不同的方式实现它。一种方法是使用ArrayList

ArrayList<Integer> set[] = new ArrayList[max];
//Initialize each set, add element into set
for(int i = 0; i < max; i++)
   set[i] = new ArrayList();
   set[i].add(i);
//For step 2
set[boss[u]].addAll(set[boss[v]]);

最后一点是，你的第4步是错的，对于你当前的实现，你只需将每个元素添加到set [boss [u]]中，但是在这一步中，它们只需要那些set [boss [boss] v]]。所以：

        int bossV = boss[v];//This is important! Answer why this is needed by yourself.
        for(int z=0;z<_boss.length;z++)
        {
            if(boss[z] == bossV)
               _boss[z]=_boss[u];
        }

小注意：您使用的这个版本不是union-find的最快版本，请尝试阅读更多相关内容！

实现Union-Find算法混淆

1 个答案: