并查集的应用——小鑫的城堡

小鑫的城堡（原题链接）

Time Limit: 1000 ms Memory Limit: 65536 KiB

Submit Statistic

Problem Description

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

Input

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

Output

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

Sample 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

Sample Output

Yes
No

Hint

 

 

Source

xj

这道题利用并查集来判断一个图中是否存在环。

到输入两个点时，那么这两个点一定是连接的，接着判断这两个点的根节点是否一样，如果一样很显然这就成了一个环不符题意然后让flag=0；如果一个点是第一次加图中要做好标记即令a[u]=1;k++;如果a[u]==1;表示点U在图中，k则表示当前有几部分（即几个集合）。

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

int map[100050], a[100050];
int find(int x)
{
    if(map[x] == x)
        return x;
    else
        return find(map[x]);
}

void merge(int u, int v)
{
    int t1 = find(u);
    int t2 = find(v);
    if(t1!=t2)
        map[t1] = t2;
}

int main()
{
    int m, u, v, i;
    while(~scanf("%d",&m))
    {
        memset(a, 0, sizeof(a));
        int flag = 1, k = 0;
        for(i=1; i<=100000; i++)
            map[i] = i;
        while(m--)
        {
            scanf("%d %d",&u,&v);
            if(a[u] == 0)
            {
                a[u] = 1;
                k++;
            }
            if(a[v] == 0)
            {
                a[v] = 1;
                k++;
            }
            if(find(u) == find(v))
                flag = 0;
            else
                k--;
            merge(u,v);
        }
        if(flag==1 && k==1)
            printf("Yes\n");
        else
            printf("No\n");
    }
    return 0;
}