HDU 1272 拓撲排序

#include<iostream>
#include<cstring>
#include<cstdio>
#include<queue>
using namespace std;

int p[100005];
int r[100005];

bool fl[100005];

int yes;

int node_count;
int edge_count;

void init()
{
    for(int i=1;i<=100000;i++)
    {
        p[i]=i;
        r[i]=1;
    }
    yes=1;
    memset(fl,0,sizeof(fl));
    node_count=0;
    edge_count=0;
}

int get(int v)
{
    if(v!=p[v])p[v]=get(p[v]);
    return p[v];
}

bool Union(int a,int b)
{
    int x=get(a);
    int y=get(b);
    if(x==y)
    {
        return 0;
    }

    if(r[x]<r[y])
    {
        p[x]=y;
    }
    else if(r[x]>r[y])
    {
        p[y]=x;
    }
    else
    {
        p[x]=y;
        r[y]++;
    }

    return 1;
}

int main()
{
    int a,b;
    init();
    while(scanf("%d%d",&a,&b))
    {
        if(a==-1&&b==-1)
        {
            break;
        }
        else if(a==0&&b==0)
        {
            if((yes&&edge_count==node_count-1)||(edge_count==0&&node_count==0))
            {
                printf("Yes\n");
            }
            else
            {
                printf("No\n");
            }
            init();
        }
        else if(a==b)
        {
            if(!fl[a])
            {
                fl[a]=1;
                node_count++;
            }
            continue;
        }
        else
        {
            edge_count++;
            if(!fl[a])
            {
                fl[a]=1;
                node_count++;
            }
            if(!fl[b])
            {
                fl[b]=1;
                node_count++;
            }
            if(!Union(a,b))
            {
                yes=0;
            }
        }
    }
    return 0;
}