Union-Find 按大小求并算法

#include

#include

using namespace std;

class UF{

  public:

    UF(int size): vec(size), parent(size) {

      for (int i = 0; i < size; i++) {

        vec[i] = i;

        parent[i] = -1;

      }

    }

    void Union(int p, int q) {

      int root1 = Find(p);

      int root2 = Find(q);

      //将小树链接到大树，大小则由数的结点个数确定

      if (root1 > root2) { // 例如-2 > -3 说明要将root1链接到root2上面

        parent[root2] += parent[root1]; //更新大树的结点个数

        parent[root1] = root2; //将树1链接到树2

      } else {

        parent[root1] += parent[root2];

        parent[root2] = root1;

      }

    }

    int Find(int x) {

      if (parent[x] < 0) //递归直到x的父节点为负数

        return x;

      else

        return parent[x] = Find(parent[x]); //带有路径压缩功能，若本次找到组号，那么递归找到组号后直接赋值给parent[x],下次查找则能在O(1)内完成

  }

    int count(int x) {

      int c = parent[Find(x)];

      return -c;

    }

  private:

    vector vec;

    vector parent; //parent有两个作用，一个是指示集合中的个数，另外若parent[x]为负数，x则为集合   //中的组号

};

int main() {

  UF uf(10);

  uf.Union(2, 3);

  uf.Union(6, 7);

  uf.Union(0, 1);

  uf.Union(1, 2);

  uf.Union(3, 6);

  cout << uf.Find(0) << endl;

  cout << uf.Find(1) << endl;

  cout << uf.Find(2) << endl;

  cout << uf.Find(3) << endl;

  cout << uf.Find(6) << endl;

  cout << uf.Find(7) << endl;

  cout << uf.count(0) << endl;

  cout << uf.count(1) << endl;

  cout << uf.count(2) << endl;

  cout << uf.count(3) << endl;

  cout << uf.count(6) << endl;

  cout << uf.count(7) << endl;

  return 0;

}

未完待续

更详细请参见 这里