老实说一开始并没有看出来是并查集。。。。感觉像是一个DFS搜索,然后跳过度为1的点,判环的办法都想好了。。。感觉应该也能写过去吧。。
但是听到别人说是并查集以后稍稍吃了一惊,因为我并没有这样的想法。。可是按照并查集的思路想了几十秒之后就明白了。。果然思想是很重要的。。
判断成环就是判断两个房间的根节点是不是同一个节点,判断是不是连通图维护一个root变量记录当前的根节点的个数就可以了,因为新的一组房间进来的时候情况是有限的。
要么两个房间都没有出现过,此时根节点增加1,一个出现过一个未出现,此时根节点不变化,假若两个都出现过,且根节点不同的话,那么根节点个数减一,假若根节点相同的话,那么此时就有环了,可以出答案了。
以下是代码:
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
int x,y,sets[100010],sz[100010],maxm,root;
int _find(int x){return (x==sets[x])?x:sets[x]=_find(sets[x]);}
bool _merge(int x,int y),flag=true;
int main(){
while(scanf("%d%d",&x,&y)&&y!=-1){
if(x!=0&&y!=0){
_merge(x,y);
while(scanf("%d%d",&x,&y)&&y){
maxm=max(maxm,max(x,y));
if(_merge(x,y)){
flag=false;
break;
}
}
if(!flag)
while(scanf("%d%d",&x,&y)&&y);
}
else{
printf("Yes\n");
continue;
}
printf("%s\n",(flag&&root==1)?"Yes":"No");flag=true;
memset(sets,0,sizeof(int)*(1+maxm));maxm=0;root=0;
}
return 0;
}
bool _merge(int x,int y){
if(sets[x]==0){
if(sets[y]==0){
sets[y]=x,sets[x]=x;
++root;
}
else
sets[x]=_find(y);
}
else{
if(sets[y]==0)
sets[y]=_find(x);
else{
int a=_find(x),b=_find(y);
if(a==b)return true;
sets[b]=a;
--root;
}
}
return false;
}