并查集

http://162.105.81.212/JudgeOnline/problem?id=1182

http://162.105.81.212/JudgeOnline/problem?id=1161

http://162.105.81.212/JudgeOnline/problem?id=2524

并查集的应用

 

并查集学习：

l         并查集：(union-find sets)

一种简单的用途广泛的集合. 并查集是若干个不相交集合，能够实现较快的合并和判断元素所在集合的操作，应用很多，如其求无向图的连通分量个数等。最完美的应用当属：实现Kruskar算法求最小生成树。

l         并查集的精髓（即它的三种操作，结合实现代码模板进行理解）：

1、Make_Set(x) 把每一个元素初始化为一个集合

初始化后每一个元素的父亲节点是它本身，每一个元素的祖先节点也是它本身（也可以根据情况而变）。

2、Find_Set(x) 查找一个元素所在的集合

查找一个元素所在的集合，其精髓是找到这个元素所在集合的祖先！这个才是并查集判断和合并的最终依据。
判断两个元素是否属于同一集合，只要看他们所在集合的祖先是否相同即可。
合并两个集合，也是使一个集合的祖先成为另一个集合的祖先，具体见示意图

3、Union(x,y) 合并x,y所在的两个集合

合并两个不相交集合操作很简单：
利用Find_Set找到其中两个集合的祖先，将一个集合的祖先指向另一个集合的祖先。如图

l         并查集的优化

1、Find_Set(x)时 路径压缩
寻找祖先时我们一般采用递归查找，但是当元素很多亦或是整棵树变为一条链时，每次Find_Set(x)都是O(n)的复杂度，有没有办法减小这个复杂度呢？
答案是肯定的，这就是路径压缩，即当我们经过"递推"找到祖先节点后，"回溯"的时候顺便将它的子孙节点都直接指向祖先，这样以后再次Find_Set(x)时复杂度就变成O(1)了，如下图所示；可见，路径压缩方便了以后的查找。

 

2、Union(x,y)时 按秩合并
即合并的时候将元素少的集合合并到元素多的集合中，这样合并之后树的高度会相对较小。

l         主要代码实现

int father[MAX]; /**//* father[x]表示x的父节点*/ int rank[MAX]; /**//* rank[x]表示x的秩*/ /**//* 初始化集合*/ void Make_Set(int x) { father[x] = x; //根据实际情况指定的父节点可变化 rank[x] = 0; //根据实际情况初始化秩也有所变化 } /**//* 查找x元素所在的集合,回溯时压缩路径*/ int Find_Set(int x) { if (x != father[x]) { father[x] = Find_Set(father[x]); //这个回溯时的压缩路径是精华 } return father[x]; } /**//* 按秩合并x,y所在的集合 下面的那个if else结构不是绝对的，具体根据情况变化 但是，宗旨是不变的即，按秩合并，实时更新秩。 */ void Union(int x, int y) { x = Find_Set(x); y = Find_Set(y); if (x == y) return; if (rank[x] > rank[y]) { father[y] = x; } else { if (rank[x] == rank[y]) { rank[y]++; } father[x] = y; } }   

注：学习并查集时非常感谢Slyar提供的资料，这里注明链接：http://www.slyar.com/blog/
另外，我认为写并查集时涉及到的路径压缩，最好用递归，一方面代码的可读性非常好，另一方面，可以更直观的理解路径压缩时在回溯时完成的巧妙。

 

poj_1161:

#include<iostream> using namespace std; int root[30000],rank[30000]; void Makeset(int n) { for(int i=0;i<n;i++) { root[i]=i; rank[i]=1; } } inline int Findset(int x) { if(x!=root[x]) root[x]=Findset(root[x]); return root[x]; } void Unionset1(int x,int y) { if(x==y) return; if(rank[x]<=rank[y]) { root[x]=y; rank[y]+=rank[x]; } else { root[y]=x; rank[x]+=rank[y]; } } int main() { int n,m,sum,i,j,k,x,y,t; while(scanf("%d%d",&n,&m)) { if(n==0 && m==0) break; Makeset(n); for(i=0;i<m;i++) { scanf("%d%d",&k,&t); for(j=0;j<k-1;j++) { scanf("%d",&y); x=Findset(t); y=Findset(y); Unionset1(x,y); } } sum=Findset(0); printf("%d/n",rank[sum]); } return 0; }

 

 

poj_1182:

#include<iostream> using namespace std; struct point{ int parent; int kind; }ufind[50010]; void init(int n) { for(int i=1;i<=n;i++) { ufind[i].parent=i; ufind[i].kind=0; } } int find(int index) { int temp; if(index==ufind[index].parent) return index; temp=ufind[index].parent; ufind[index].parent=find(ufind[index].parent); ufind[index].kind=(ufind[temp].kind+ufind[index].kind)%3; return ufind[index].parent; } void unions(int rootx, int rooty, int x, int y, int dkind) { ufind[rooty].parent=rootx; ufind[rooty].kind=(-dkind+(ufind[x].kind-ufind[y].kind)+3)%3; } int main() { int n,k,count=0,d,x,y; scanf("%d%d",&n,&k); init(n); while(k--) { scanf("%d%d%d",&d,&x,&y); if(x>n || y>n) count++; else if(d==2 && x==y) count++; else { int rx=find(x); int ry=find(y); if(rx!=ry) unions(rx,ry,x,y,d-1); else { if(d==1 && ufind[x].kind!=ufind[y].kind) count++; if(d==2 && (ufind[x].kind-ufind[y].kind+3)%3!=1) count++; } } } printf("%d/n",count); return 0; }

 

 

poj_2524:

#include<iostream> using namespace std; int father[50001], rank[50001]; void Makeset(int n) { for(int i=1;i<=n;i++) { father[i]=i; rank[i]=0; } } inline int Findset(int x) { if(x!=father[x]) father[x]=Findset(father[x]); return father[x]; } void Unionset(int x,int y) { x=Findset(x); y=Findset(y); if(rank[x]>rank[y]) father[y]=x; else { if(rank[x]==rank[y]) rank[y]++; father[x]=y; } } int main() { int n,m,sum,a,b,id=1; while(scanf("%d%d",&n,&m)) { if(n==0 && m==0) break; sum=n; Makeset(n); while(m--) { scanf("%d%d",&a,&b); if(Findset(a)!=Findset(b)) { sum--; Unionset(a,b); } } printf("Case %d: %d/n",id++,sum); } return 0; }

 

poj 2492(与1703非常相似)

#include<iostream> using namespace std; #define MAX 2001 int father[MAX]; int opp[MAX]; bool flag; void initSet(int n) { flag =false; for(int i=1; i<=n; i++) { father[i] = i; opp[i] = -1; } } int findSet(int x) { if(x != father[x]) father[x] = findSet(father[x]); return father[x]; } void unionSet(int x, int y) { x = findSet(x); y = findSet(y); father[x] = y; } int main() { //freopen("in.txt", "r", stdin); int t, n, m, a, b, id = 1; scanf("%d", &t); while(t--) { scanf("%d %d", &n, &m); initSet(n); while(m--) { scanf("%d%d", &a, &b); if(!flag) { if(findSet(a) == findSet(b)) flag = true; else { if(opp[a] == -1 && opp[b] == -1) { opp[a] = b; opp[b] = a; } else if(opp[a] != -1 && opp[b] != -1) { unionSet(a, opp[b]); unionSet(b, opp[a]); } else if(opp[a] != -1) unionSet(b, opp[a]); else if(opp[b] != -1) unionSet(a, opp[b]); } } } printf("Scenario #%d:/n", id++); if(flag) puts("Suspicious bugs found!/n"); else puts("No suspicious bugs found!/n"); } return 0; }

 

poj 1703

#include<iostream> using namespace std; #define MAX 100001 int father[MAX]; int opp[MAX]; void initSet(int n) { for(int i=1; i<=n; i++) { father[i] = i; opp[i] = -1; } } int findSet(int x) { if(x < 0) return -2; if(x != father[x]) father[x] = findSet(father[x]); return father[x]; } void unionSet(int x, int y) { x = findSet(x); y = findSet(y); father[x] = y; } int main() { //freopen("in.txt", "r", stdin); int t, n, m, a, b; char op; scanf("%d", &t); while(t--) { scanf("%d%d", &n, &m); initSet(n); while(m--) { scanf("%*c%c%d%d", &op, &a, &b); if(op == 'A') { if(a == b || findSet(a) == findSet(b)) puts("In the same gang."); else { if(findSet(opp[a]) == findSet(b)) puts("In different gangs."); else puts("Not sure yet."); } } else if(op == 'D') { if(opp[a] == -1 && opp[b] == -1) { opp[a] = b; opp[b] = a; } else if(opp[a] != -1 && opp[b] != -1) { unionSet(a, opp[b]); unionSet(b, opp[a]); } else if(opp[a] != -1) { unionSet(b, opp[a]); opp[b] = a; } else if(opp[b] != -1) { unionSet(a, opp[b]); opp[a] = b; } } } } return 0; }