小希的迷宫(并查集)

http://acm.hdu.edu.cn/showproblem.php?pid=1272

思路:

所求的图满足两点要求：(1) 连通 (2) 不存在环 。

而这两点用并查集都能判断。注意本题中 输入 0 0时，应输出 Yes。

#include <stdio.h>
#include <string.h>
const int N=100002;
int f[N],vis[N];
void init()
{
    for (int i = 1; i <= N; i ++)
    {
        f[i] = i;
        vis[i] = 0;
    }

}
int find(int x)
{
    if (x!=f[x])
        f[x] = find(f[x]);
    return f[x];
}
void merge(int x,int y)
{
    x = find(x);
    y = find(y);
    if (x!=y)
        f[x] = y;
}
int main()
{
    int u,v;
    while(1)
    {
        int cnt = 0;
        int flag = 1;
        init();
        while(~scanf("%d%d",&u,&v))
        {
            if (u==0&&v==0||u==-1&&v==-1)
                break;
            vis[u] = 1;
            vis[v] = 1;
            if(find(u)==find(v))//判环
            {
                flag = 0;
            }
            else
                merge(u,v);
        }
        if (u==-1&&v==-1)
            break;
        for (int i = 0; i <= N; i ++)//判连通性
        {
            if (vis[i]&&f[i]==i)
                cnt ++;
        }
        if (cnt<=1&&flag)
            printf("Yes
");
        else
            printf("No
");
    }
    return 0;
}