HDU 1272

WA好多次，注意题目，多颗树算No。注意判断树数量的代码。

以我目前的智商，暂时只能用这个拙劣的方法，判断集合的连通性，也就是节点是否只构成一棵树。

#include<cstdio>
#include<cstring>
int node[100001];
int tree=0;
int son[100001];
void init()
{
    for(int i=1;i<=100000;i++)
        node[i]=i;
}

int find(int a)
{
    if(node[a]!=a)
    {
        node[a]=find(node[a]);
    }
    return node[a];
}

bool unin(int a,int b)
{
    int ai=find(a);
    int bi=find(b);
    if(ai==a&&bi==b&&son[a]==0&&son[b]==0) tree++;
    if((ai!=a&&bi!=b&&ai!=bi)||(son[ai]>0&&son[bi]>0&&ai!=bi)) tree--;

    //printf("tree=%d\n",tree);
    if(ai==bi) { return false;}
    else {node[ai]=bi; son[bi]++; return true;}
}

int main()
{
    int a,b;
    bool bed=0;
    bool first=1;
    init();
    memset(son,0,sizeof(son));
    while(scanf("%d%d",&a,&b)!=EOF)
    {
        if(a==-1&&b==-1) break;
        if(first==1&&a==0&&b==0) {printf("Yes\n"); continue;}
        if(a==0&&b==0)
        {
            if(bed==1||tree!=1) printf("No\n");
            else printf("Yes\n");
            bed=0;
            tree=0;
            first=1;
            init();
            memset(son,0,sizeof(son));
        }
        else if(a==b) continue;
        else
        {
            if(unin(a,b)==0) bed=1;
            first=0;
        }
    }
    return 0;
}