【HDU】1272小希的迷宫（并差集判断连通且结构为树）

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=1272

题目：

上次Gardon的迷宫城堡小希玩了很久（见Problem B），现在她也想设计一个迷宫让Gardon来走。但是她设计迷宫的思路不一样，首先她认为所有的通道都应该是双向连通的，就是说如果有一个通道连通了房间A和B，那么既可以通过它从房间A走到房间B，也可以通过它从房间B走到房间A，为了提高难度，小希希望任意两个房间有且仅有一条路径可以相通（除非走了回头路）。小希现在把她的设计图给你，让你帮忙判断她的设计图是否符合她的设计思路。比如下面的例子，前两个是符合条件的，但是最后一个却有两种方法从5到达8。

思路：

如果给的图是一个树(且输入中不能有重复边输入。如果有，也应该输出No)，那么输出Yes，否则输出No。

        一个树满足下面两个条件：只有1个连通分量且任意两点间不存在两条路。

        判断任意两点间是否存在两条路只要看已经连通的两点是否还会插入另外一条边进去？

        注意:对于输入数据0 0需要特殊判断。
代码：

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<set>
#define maxn 100000+5
using namespace std;

int fa[maxn];
int findset(int x)
{
    return fa[x]==-1? x:fa[x]=findset(fa[x]);
}

int bind(int u,int v)
{
    int fu=findset(u);
    int fv=findset(v);
    if(fu!=fv)
    {
        fa[fu]=fv;
        return 1;
    }
    return 0;
}
int main()
{
    int u,v;
    while(scanf("%d%d",&u,&v)==2&&u>=0)
    {
        if(u==0&&v==0)
        {
            printf("Yes\n");
            continue;
        }

        memset(fa,-1,sizeof(fa));
        set<int >st;
        bool ok=true;

        do
        {
            if(!bind(u,v)) ok=false;
            st.insert(u);
            st.insert(v);
        }while(scanf("%d%d",&u,&v)==2&&u);

        if(ok)
        {
            int cnt=0;
            for(set<int>::iterator it=st.begin(); it!=st.end(); ++it)
            {
                if(*it == findset(*it) )
                    cnt++;
            }
            if(cnt>1) ok=false;
        }
        printf("%s\n",ok?"Yes":"No");
    }
    return 0;
}