小希的迷宫 HDU - 1272

#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
const int N=1e5;
int a,b;
int p[N+];
// 标记走过的房间
bool vis[N+];
int find(int x)
{
    if(p[x]!=x)
        p[x]=find(p[x]);
    return p[x];
}
void Union(int a,int b)
{
    int pa=find(a);
    int pb=find(b);
    if(pa!=pb)
        p[pb]=pa;
}
int main()
{
    while(cin>>a>>b)
    {
        if(a==-&&b==-)
            break;
        // a，b 均为 0 时，也要输出 Yes；空树也是树
        if(a==&&b==)
        {
            cout<<"Yes"<<endl;
            continue;
        }
        for(int i=;i<=N;i++)
            p[i]=i,vis[i]=;
        int maxx=;
        bool flag=;
        while(a||b)
        {
            vis[a]=;
            vis[b]=;
            //是否成环
            if(find(a)==find(b))
                flag=;
            Union(a,b);
            // 找出房间编号的最大值
            maxx=max(maxx,max(a,b));
            cin>>a>>b;
        }
        if(flag)
            cout<<"No"<<endl;
        else
        {
            int cnt=;
            // 计算共有几个符合条件的联通集合
            for(int i=;i<=maxx;i++)
                if(vis[i]&&p[i]==i)
                    cnt++;
            // 这里可能存在不止一个集合，这样的话 ans 值就大于 1 了，这样就不满足题意了，就该输出 No
            if(cnt==)
                cout<<"Yes"<<endl;
            else
                cout<<"No"<<endl;
        }
    }
    return ;
}