笔试算法题（38）：并查集（Union-Find Sets）

 1 int *father;
 2 int *rank;
 3 
 4 /**
 5  * 并查集的初始化：
 6  * 数组father中的元素在最开始ide时候都是独立的树，也就是只有根节点
 7  * 的树，数组father的下标i表示节点，而father[i]的值表示i节点的父亲
 8  * 节点；rank[i]=1表示一开始所有树节点的高度都为1
 9  * */
10 void init(int cap) {
11         father=new int[cap];
12         rank=new int[cap];
13         /**
14          * 时间复杂度为O(N)
15          * */
16         for(int i=0;i<10;i++) {
17                 father[i]=i;
18                 rank[i]=1;
19         }
20 }
21 
22 void clean() {
23         delete [] father;
24         delete [] rank;
25 }
26 /**
27  * 查找元素所在的集合并进行路劲压缩：
28  * 由于需要频繁使用GetFather（）函数，并且其时间复杂度受树结构影响；
29  * 当元素较多的时候，集合退化成链表，则GetFather()需要O(N)，所以
30  * 需要对其进行优化，每次调用GetFather（）的时候都将输入元素压缩成
31  * 最原始父亲节点的直接子节点
32  * */
33 int GetFather(int son) {
34         if(father[son]==son)
35                 return son;
36         else {
37                 father[son]=GetFather(father[son]);
38                 return father[son];
39         }
40 }
41 
42 /**
43  * 合并两个不相交的集合:
44  * 输入元素x和y来自两个不相交的集合，找到其最原始的父亲节点
45  * 并将一个原始父亲节点设置为另一个原始父亲节点的父亲节点
46  * */
47 void Union1(int x, int y) {
48         /**
49          * GetFather()为递归寻找输入节点的最原始的父亲节点
50          * */
51         int fx=GetFather(x);
52         int fy=GetFather(y);
53         /**
54          * 判断x和y是否来自同一棵树，如果不是才进行赋值；其实可以
55          * 不同进行判断（省去if语句）
56          * 注意最原始父亲节点的father[i]=i;
57          * */
58         if(fx!=fy)
59                 father[fx]=fy;
60 }
61 
62 /**
63  * 利用rank加权数组启发式进行合并
64  * */
65 void Union2(int x, int y) {
66         int fx=GetFather(x);
67         int fy=GetFather(y);
68 
69         if(fx==fy) return;
70         /**
71          * rank[fx]较大，说明其越靠近根节点，则将
72          * fy连接到其后面可以压缩路径
73          * */
74         if(rank[fx]>rank[fy])
75                 father[fy]=fx;
76         else {
77                 if(rank[fx]==rank[fy])
78                         rank[fy]++;
79                 father[fx]=fy;
80         }
81 }
82 
83 /**
84  * 判断两个元素是否属于同一个集合：
85  * 利用GetFather（）函数判断其最原始父亲节点是否相同
86  * */
87 bool IsSameSet(int x, int y) {
88         return GetFather(x)==GetFather(y);
89 }