3 3 1 2 2 3 3 1 3 3 1 2 2 3 3 2 0 0
Yes No
炸一看知道是并查集~再一看还知道是并查集,但是要咋写呢~
这里直接上实例:
一般的并查集的题目可以理解为单源连通图,比如并查集数组pre[]中存数据如下:3 3 3,我们知道 点1的祖先是3号,1->...->3,同理,2号的祖先也是3号2->.....->3无论中间是否有繁杂的路径,我们知道1 2 的祖先是3,但是3的祖先一定不可能是1 2 。
本题题意是说,所有的房间都连通,如果说1 2 3 4 5 6....n-1的找到的祖先都是n并且反过头说,n也能找到1 2 3 4 5 6...n-1我们就构成了这个条件:所有房间都连通.
因为我们知道一维的并查集数组可以完成单向完成查找,如果反过去也能找的话就能说所有的房间都连通了.
这里先贴上一维并查集数组的merge操作:
void merge(int a,int b) { int A,B; A=find(a); B=find(b); if(A!=B) f[B]=A;//f[A]=B; }这里我们只能完成B->A或者是A->B 的操作,如果我们能同时操作就完美的实现了刚才的理论:用两个数组来实现.
void merge(int a,int b) { if(a!=n) { int fa=find(a,0),fb=find(b,0); if(fa!=fb) pre[0][a]=b;//这里应用了二维数组.表示有两个并查集数组. } if(b!=n) { int fa=find(a,1),fb=find(b,1); if(fa!=fb) pre[1][b]=a; } }
正向查找到n的同时逆向也能查找到n就满足了环,同时满足所有房间都连通。然后贴上完整AC代码:
#include<stdio.h> #include<string.h> using namespace std; int n,m; int pre[2][100010]; int find(int a,int i) { int r=a; while(r!=pre[i][r]) { r=pre[i][r]; } return r; } void merge(int a,int b) { if(a!=n) { int fa=find(a,0),fb=find(b,0); if(fa!=fb) pre[0][a]=b; } if(b!=n) { int fa=find(a,1),fb=find(b,1); if(fa!=fb) pre[1][b]=a; } } int main() { while(scanf("%d%d",&n,&m)!=EOF,n||m) { if(n==0||m==0)break; for(int i=0;i<=n;i++) { pre[0][i]=pre[1][i]=i; } while(m--) { int a,b; scanf("%d%d",&a,&b); merge(a,b); } int w=1; /*for(int i=1;i<=3;i++) { printf("%d ",pre[0][i]); } for(int i=1;i<=3;i++) { printf("%d ",pre[1][i]); }*/ for(int i=1;i<=n;i++) { // printf("%d %d\n",find(i,0),find(i,1)); if(find(i,0)!=n||find(i,1)!=n) { w=0; break; } } if(w) printf("Yes\n"); else printf("No\n"); } }