hdu 1272 并查集

hdu 1272 小希的迷宫

题意：判断图是否连通且无环；

思路：输入边的两点，若两点的父节点相同，则存在环；只有一个根节点则图连通；当输入"0 0"时输出"Yes"

#include<cstdio>
#include<cstring>

int set[100000 + 50];
bool visit[100000 + 50];
bool flag;
int find(int x)
{
    return x == set[x]?x:set[x] = find(set[x]);
}

void merge(int a, int b)
{
     int i = find(a);
     int j = find(b);
     if(i != j) set[i] = j;
     else flag = true;
     visit[a] = visit[b] = true;
     return ;
}

int main()
{
    int a,b;
    while(~scanf("%d%d",&a,&b))
    {
        if(a == -1 && b == -1)   break;
        else if(a == 0 && b == 0)
        {
            printf("Yes
");
            continue;
        }
        else
        {
            for(int i = 1; i <= 100000; i++)
                set[i] = i;
            merge(a,b);
        }
        while(scanf("%d%d",&a,&b) && (a||b))
        {
            merge(a,b);
        }
        int num = 0;
        for(int i = 1; i <= 100000; i++)
        {
             if(visit[i] && set[i] == i) num++;
        }
        if(!flag && num == 1) printf("Yes
");
        else printf("No
");
        flag = false;
        memset(visit,0,sizeof(visit));
    }
    return 0;
}