並查集的樹形實現(C++)（原創）

摘要：本文介紹了通用並查集的樹形實現，通過壓縮路徑和維持數的平衡，可以保證

查找和合並的平均時間複雜度為O(1)!

關鍵字：並查集，UnionFind，樹形

並查集基本知識參見博文《並查集的數組實現》。在並查集的樹形實現中，使用樹狀結構來組織一個

集合，多個集合之間組成森林。

合並兩個集合即合並兩棵樹T1和T2時，隻要將兩棵樹中的一棵作為另外一棵的子樹。為了維護數的

平衡，即減少深度較深的節點的個數，可以把節點較少的數作為節點較深的樹的子樹。具體實現就是把一棵

樹的根節點作為另外一棵樹的根節點的子樹！

查找某個節點所屬的集合時，找到該節點所在樹的樹根即可，同時，可以將查找經過的所有節點的父

節點都設置為根，以方便下一次的查詢，這種策略雖然耗費了時間，但是可以使查找的平均時間複雜度

為O（1）！

基於以上思想的並查集樹形實現C++源碼如下：

/*
 *並查集的樹形實現
 */

#include <iostream>

using namespace std;

#define N 10000

struct Node
{
      int parent; //父節點編號
      int count; //集合中元素的個數
};

Node UnionFindSet[N];

void uf_init(int n); //初始化
int uf_find(int x); //查找元素，返回集合根節點編號
void uf_merge(int x, int y); //合並元素所在的集合樹

int main(int argc, char* *argv)
{
      int n = 20;
      uf_init(n);
      uf_merge(1, 5);
      uf_merge(10, 13);
      uf_merge(15, 17);
      uf_merge(1, 13);
      uf_merge(5, 17);
      uf_merge(19, 18);
      if(uf_find(1) == uf_find(15))
      {
            cout<<"success1"<<endl;
      }
      if(uf_find(13) == uf_find(17))
      {
            cout<<"success2"<<endl;
      }
      if(uf_find(19) != uf_find(17))
      {
            cout<<"success3"<<endl;
      }
      system("pause");
      return 0;
}

void uf_init(int n)
{
      //時間複雜度O(N)
      for(int i = 0; i < n; ++i)
      {
            UnionFindSet[i].count = 1;
            UnionFindSet[i].parent = i;
      }
}

int uf_find(int x) //x >= 0 && x <n
{
      //使用路徑壓縮之後的平均時間複雜度為O(1)
      int i = x;
      while(i != UnionFindSet[i].parent)
            i = UnionFindSet[i].parent;

      //壓縮路徑，將尋找X時經過的所有節點的父節點設為集合的根節點，方便下次查找
    int j = x;
      while(j != i)
      {
            int tmp = UnionFindSet[j].parent;
            UnionFindSet[j].parent = i;
            j = tmp;
      }
      return i;
}

void uf_merge(int x, int y)
{

      //平均時間複雜度為O(1);

      //尋找樹根(集合表示符)
      x = uf_find(x);
      y = uf_find(y);
      if(y == x) return;

    //將小樹掛到大的樹下麵，可以減少較深節點的數目，維持樹的平衡,以便高效查詢
      if(UnionFindSet[x].count > UnionFindSet[y].count)
      {
            UnionFindSet[y].parent = x;
            UnionFindSet[x].count += UnionFindSet[y].count;
      }
      else
      {
            UnionFindSet[x].parent = y;
            UnionFindSet[y].count += UnionFindSet[x].count;
      }

}
//參考資料：http://www.docin.com/p-46207828.html