小鑫的城堡

SDUTOJ2798

小鑫的城堡

Time Limit: 1000MS Memory Limit: 65536KB
Submit Statistic
Problem Description

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

Input

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

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

5
2 5
2 3
1 3
3 6
4 6
6
1 2
1 3
3 4
3 5
5 6
6 4
Example Output

Yes
No
Hint

Author

xj

并查集，寻找任意两点之间是否只存在一条路径

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#define MAXN 123456
int pre[MAXN], book[MAXN];
int root(int x)//寻找祖先
{
    while(pre[x]!=x)
        x = pre[x];
    return x;
}
void merge(int a, int b)//合并
{
    if(root(a)!=root(b))
        pre[root(a)] = root(b);
}
int main()
{
    int m, a, b;
    while(~scanf("%d", &m))
    {
        memset(book, 0, sizeof(book));//标记数组
        for(int i=0;i<=MAXN;i++)//初始化
            pre[i] = i;
        for(int i=0;i<m;i++)
        {
            scanf("%d %d", &a, &b);
            merge(a, b);//合并
            book[a] = 1;
            book[b] = 1;
        }
        int sum1 = 0, sum2 = 0;
        for(int i=1;i<=MAXN;i++)
        {
            if(book[i]==1)//查看是否所有的点都已经出现过
                sum1++;
            if(book[i]==1&&pre[i]==i)//有几个祖先
                sum2++;
        }
        if(sum2==1&&sum1==m+1)//只有一个祖先并且所有的点都出现过
            printf("Yes\n");
        else
            printf("No\n");
    }
    return 0;
}