题目:中文的~~~~
思路:属于并查集算法,输出YES的条件有两个,第一:每次新输入的两个数不能同属于一个集合(即根节点一样),第二:所有的输入完成后判断是否仅有一个集合(根节点只有一个)。只有这两个条件全部达成,才能输出“YES”~~~需要注意的是 直接输入“0 0”,应该输出“YES”。
#include <iostream>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <cmath>
#include <cctype>
#include <algorithm>
using namespace std;
const int MAXN = 1e5 + 3;
int pre[MAXN];
bool bol[MAXN];
int maxn = -1;
int flag = 1; //标记是否出现了根节点一样的两个点、即条件一、默认是没有出现
int Find(int x)
{
int r = x;
while(pre[r] != r)
{
r = pre[r];
}
int i = x,j;
while(pre[i] != r)
{
j = i;
i = pre[i];
pre[j] = r;
}
return r;
}
void Mix(int a,int b)
{
int x = Find(a);
int y = Find(b);
if(x > y)
{
pre[x] = y;
}
if(x < y)
{
pre[y] = x;
}
}
void Mst() //初始化
{
maxn = -1;
flag = 1;
for(int i = 1; i <= MAXN; i++)
{
pre[i] = i;
bol[i] = false;
}
}
int Search(int maxn) //计算集合的个数
{
int cnt = 0;
for(int i = 1; i <= maxn; i++)
{
if(bol[i])
if(pre[i] == i)
cnt++;
}
if(maxn <= 0)
return 1;
return cnt;
}
int main()
{
int M,N;
Mst();
while(~scanf("%d%d",&M,&N)&&(M != -1 || N != -1))
{
maxn = max(max(M,N),maxn);
bol[M] = true; //这个点出现过
bol[N] = true;
if(M||N)
{
int m = Find(M);
int n = Find(N);
if(flag && m == n)
flag = 0; //不满足条件一
else
Mix(m,n);
}
else
{
int single = Search(maxn); //是否只有一个根节点
if(single== 1 && flag) printf("Yes\n"); //两个条件都满足
else printf("No\n");
Mst();//注意数据初始化
}
}
return 0;
}