hdu-1272 小希的迷宫

#include <iostream>
using namespace std;
int size[100010],tree[100010],vis[100010];
bool flag=true;
void makeSet(int x)
{
    size[x]=1;
    tree[x]=x;
    vis[x]=0;
}
int findSet(int x)
{
    if(x!=tree[x])
        tree[x]=findSet(tree[x]);
    return tree[x];
}
void Union(int x,int y)
{
    int fx=findSet(x);
    int fy=findSet(y);
    if(fx==fy)
    {
        flag=false;
        return;    
    }
    if(size[fx]<size[fy])            
    {
        tree[fx]=fy;
        size[fy]+=size[fx];
    }
    else
    {
        tree[fy]=fx;
        size[fx]+=size[fy];
    }
}
int main(void)
{
    int a,b;
    for(int i=0;i<100010;i++)
        makeSet(i);
    while(cin>>a>>b&&(a!=-1&&b!=-1))
    {
        if(a==0&&b==0)
        {
            int k=0;
            for(int i=1;i<100010;i++)
                if(tree[i]==i&&vis[i]!=0)
                    k++;
            if(k>1)
                flag=false;
            if(flag)
                cout<<"Yes"<<endl;
            else
                cout<<"No"<<endl;
            for(int i=0;i<100010;i++)
                makeSet(i);
            flag=true; 
        }
        if(a!=b)
        {
            Union(a,b);
            vis[a]=1;
            vis[b]=1;
        }    
    }
    return 0;
}