这题是我在某场比赛时看到的,以为并查集可以搞定啊,结果花式LTE,各种WA,后来看了题解都没搞懂为何我那么写就会TLE。这题有几个需要注意的地方,首先就是给你一些路径,让你判断这个图中有没有环,如果有环就输出No,还有就是给你的图只能有一个根节点,如果有多个根,也输出No。输入要放在外面就一个输入,不然如果直接输入0 0,一个没有点的图,就会WA。
#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
int pre[100005],pp[100005];
int flag;
int Find(int x)//合并操作
{
int r=x;
while(pre[r]!=r)
r=pre[r];
int i=x,j;
while(pre[i]!=r)
{
j=pre[i];
pre[i]=r;
i=j;
}
return r;
}
void mix(int x,int y)
{
int fx=Find(x);
int fy=Find(y);
if(fx!=fy) pre[fx]=fy;
else flag=1;//如果a和b已经在一个集合内了,那么就会出现环。于是就不是一条路径了。
}
int main()
{
int a,b,maxn=0,ans=0;
flag=0;
for(int i=1;i<=100000;i++) pre[i]=i;
memset(pp,0,sizeof(pp));
while(cin>>a>>b)//一定要在外面输入,如果直接输入0 0的话,就直接进行操作,如果在后面判断是否是0 0,就会WA
{
if(a==-1&&b==-1) break;
if(a>maxn) maxn=a;
if(b>maxn) maxn=b;//减少一些数据范围
if(a==0&&b==0)
{
for(int i=1;i<=maxn;i++)
{
if(pp[i]&&pre[i]==i) ans++;//ans是计算树根的个数,一个图只能由一个根
}
if(flag||ans>1) printf("No\n");
else printf("Yes\n");
flag=0;
for(int i=1;i<=100000;i++) pre[i]=i;//各种初始化=。=
maxn=0;
ans=0;
memset(pp,0,sizeof(pp));
continue;
}
mix(a,b);//合并
pp[a]=1;
pp[b]=1;
}
return 0;
}