kuangbin专题五 并查集 HDU1272 小希的迷宫

题意:
中文题
题解:
任意两个房间有且仅有一条路径可以相通,可以用并查集来做,如果输入的两个房间号的祖宗是一样的话,就说明,成环了,输出No,还有一种情况是不成环,森林的情况,比如:
1 2 3 4 0 0
正确的输出为No。
只要在最后跑一边for循环检查是否出现两个祖宗就可以知道是否出现森林了。

#include
#include
#include
using namespace std;
const int MAXN=1e5+7;
int f[MAXN];
bool vis[MAXN];
void init()
{
    for(int i=0;i<=1e5;i++)
    f[i]=i;
}
int find(int p)
{
    while(p!=f[p])
    {
        f[p]=f[f[p]];
        p=f[p];
    }
    return p;
}
bool Union(int p,int q)
{
    int P=find(p);
    int Q=find(q);
    if(P==Q)
    return true;
    else
    {
        f[Q]=P;
    }
    return false;
}
int main() 
{
    int n;
    while(1)
    {
        int a,b,MAX=0;
        init(); 
        memset(vis,false,sizeof(vis));
        bool mark=false,ans=false;
        while(~scanf("%d%d",&a,&b))
        {
            if(a==-1&&b==-1)
            {
                mark=true;
                break;
            }
            MAX=max(max(MAX,a),b);
            vis[a]=true;
            vis[b]=true;
            if(a==0&&b==0)
            break;
            if(Union(a,b))
            ans=true;
        }
        int sum=0;
        for(int i=1;i<=MAX;i++)
        if(vis[i]&&f[i]==i)
        sum++;
        if(mark)
        break;
        if(ans||sum>1)
        printf("No\n");
        else
        printf("Yes\n");
    }
}

你可能感兴趣的:(并查集)