小希的迷宫-并查集

原题链接:https://vjudge.net/problem/11138/origin

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

Input
输入包含多组数据,每组数据是一个以0 0结尾的整数对列表,表示了一条通道连接的两个房间的编号。房间的编号至少为1,且不超过100000。每两组数据之间有一个空行。
整个文件以两个-1结尾。
Output
对于输入的每一组数据,输出仅包括一行。如果该迷宫符合小希的思路,那么输出"Yes",否则输出"No"。
Sample Input
6 8 5 3 5 2 6 4
5 6 0 0

8 1 7 3 6 2 8 9 7 5
7 4 7 8 7 6 0 0

3 8 6 8 6 4
5 3 5 6 5 2 0 0

-1 -1
Sample Output
Yes
Yes
No

题解:并查集,感觉是道水题,不过我花了好长时间,这道题有个坑点,就是可能会有多个集合,这个不能忽视,要在结束时遍历一下.

#include <bits/stdc++.h>
const int N=1e5+5;
using namespace std;
int se[N];
int ma[N];
int maxx=0;
//int m[N][2];
int fd(int a){
    int x=a;
    while(se[x]!=x){
        x=se[x];
        for(int i=a,j;i!=x;i=j){
            j=se[i];
            se[i]=x;
        }

    }
    return x;
}
bool join(int a,int b){
    int fa=fd(a);
    int fb=fd(b);
    if(fa==fb) return false;
    se[fa]=fb;
    return true;
}
void wait(){
    int a,b;
    while(~scanf("%d%d",&a,&b)&&a!=0);
    printf("No\n");
    memset(ma,0,sizeof(ma));
    maxx=0;
}
int main(){
    int a,b;
    while(~scanf("%d%d",&a,&b)&&a!=-1){
        if(a==0&&b==0){
            int ans=0;
            for(int i=1;i<=maxx;i++) if(se[i]==i&&ma[i]==1) ans++;//就是这里,万恶之源,结束时要遍历看看有几个集合
            if(ans>1) printf("No\n");
            else printf("Yes\n");
            memset(ma,0,sizeof(ma));
            maxx=0;
            continue;
        }
        maxx=max(a,max(b,maxx));
        if(ma[a]==0) ma[a]=1,se[a]=a;
        if(ma[b]==0) ma[b]=1,se[b]=b;
        if(!join(a,b)) wait();
    }
}

猜你喜欢

转载自www.cnblogs.com/-yjun/p/10351297.html