HDU 1272 并查集

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

int pre[101101],f,s[101101];
int find1(int n)
{
    int t=n;
    while(pre[t]!=t)
    {
        t=pre[t];
    }
    return t;
}

void cnt(int x,int y)//建图
{
    int fx=find1(x);
    int fy=find1(y);
    if(fx!=fy)
    {
        pre[fx]=fy;//如果不在一条路上，连接起来；
    }
    else
    {
        f=1;//如果已经在一条路上了，标记；
    }
    return ;
}

int main()
{
    int n,m,a,b;
    while(~scanf("%d%d",&n,&m))
    {
        f=0;
        if(n==-1&&m==-1)
        {
            break;
        }
        for(int i=1; i<=100000; i++)
        {
            pre[i]=i;
        }
        if(n==0&&m==0)
        {
            printf("Yes\n");
        }
        else
        {
            cnt(n,m);
            int a1;
            a1=0;
            s[a1++]=n;
            s[a1++]=m;
            while(1)
            {
                scanf("%d%d",&a,&b);
                s[a1++]=a;
                s[a1++]=b;
                if(a==0&&b==0)
                {
                    break;
                }
                if(a!=b)
                {
                    cnt(a,b);
                }
            }
            if(f==1)
            {
                printf("No\n");
            }
            else
            {
                int ans=find1(n);
                int i;
                for(i=1; i<a1-2; i++)//检查连通性，是否有同一个父节点；
                {
                    int l=find1(s[i]);
                    if(l!=ans)
                    {
                        printf("No\n");
                        break;
                    }
                }
                if(i==a1-2)
                {
                    printf("Yes\n");
                }
            }
        }
    }
    return 0;
}