SDUTOJ 2798 小鑫的城堡 ——并查集

小鑫的城堡

Time Limit: 1000ms   Memory limit: 65536K  有疑问？点这里^_^

题目描述

从前有一个国王，他叫小鑫。有一天，他想建一座城堡，于是，设计师给他设计了好多简易图纸，主要是房间的连通的图纸。小鑫希望任意两个房间有且仅有一条路径可以相通。小鑫现在把设计图给你，让你帮忙判断设计图是否符合他的想法。比如下面的例子，第一个是符合条件的，但是，第二个不符合，因为从5到4有两条路径（5-3-4和5-6-4）。

输入

多组输入，每组第一行包含一个整数m（m < 100000），接下来m行，每行两个整数，表示了一条通道连接的两个房间的编号。房间的编号至少为1，且不超过100000。

输出

每组数据输出一行，如果该城堡符合小鑫的想法，那么输出"Yes"，否则输出"No"。

示例输入

5
2 5
2 3
1 3
3 6
4 6
6
1 2
1 3
3 4
3 5
5 6
6 4

示例输出

Yes
No

提示

 

来源

xj

把输入的数记录下来（hs数组是为了不记录重复的点），按照克鲁斯卡尔的思想，如果两个点已经间接连接，这两个点再连接的话必然成环，最后判所有的点是否都连接，去掉两个点集的数据，例如：

2

1 2

3 4

输出为No 

#include <stdio.h>
#include <stdlib.h>
#include <string.h>

int s[100010];
int hs[100010];
int q[100010];

int nfind(int x)
{
int r = x,q;
while(x != s[x])
{
x = s[x];
}
while(s[r] != x)
{
q = s[r];
s[r] = x;
r = q;
}
return x;
}

int main()
{
int i,n,k,x,y,bj,flag;
while(~scanf("%d",&n))
{
bj = 0;
flag = 1;
k = 0;
memset(hs,0,sizeof(hs));
for(i = 0;i <= 100000;i++)
{
s[i] = i;
}
for(i = 0; i < n;i++)
{
scanf("%d%d",&x,&y);
if(!hs[x])
{
hs[x] = 1;
q[k++] = x;
}
if(!hs[y])
{
hs[y] = 1;
q[k++] = y;
}
int fx = nfind(x);
int fy = nfind(y);
if(fx != fy)
{
s[fx] = fy;
}
else
{
flag = 0;
}
}
// for(i = 1;i < 7;i++)
// {
// printf("%d ",s[i]);
// }
// puts("");
if(flag)
{
bj = nfind(q[0]);
// printf("bj = %d\n",nfind(q[0]));
for(i = 1;i < k;i++)
{
if(nfind(q[i]) != bj)
{
flag = 0;
break;
}
}
}
if(flag)
{
printf("Yes\n");
}
else
{
printf("No\n");
}
}
return 0;
}