HDU 1272 并查集解决无向图回路

小希的迷宫

上次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<iostream>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<cmath>
#include<queue>
#include<set>
using namespace std;
typedef long long ll;
#define rep(i,a,b) for(int i=a;i<=b;i++)
const int N=1e6+7;
int m,n;
int f[N];
int a,b;
set<int> s;//存放不同的点，因为如果存的点的数目==边数+1，那么就不是一个环
//但是有一个地方会卡死（但是如果单纯不用并查集去做，只用set判断的话也是能AC）
//如果存在这样一组样例：1-2  3-4 4-5 3-5，就会判断出错，因此set仅是用于判断
//点与边的关系，判断是否存在两个集合，那么并查集的作用就是判断呢是否有环的存在
void Init(){
    rep(i,0,100000) f[i]=i;
}
int Findf(int v){
    if(f[v]==v) return f[v];
    else{
        f[v]=Findf(f[v]);
        return f[v];
    }
}
bool Union(int a,int b){
    int t1=Findf(a);
    int t2=Findf(b);
    if(t1!=t2)
        {f[t2]=t1;return false;}
    return true;
}

int main()
{
    #ifndef ONLINE_JUDGE
    freopen("in.txt","r",stdin);
    #endif // ONLINE_JUDGE
    while(scanf("%d%d",&n,&m)!=EOF){
    s.clear();
    int sum=0;//记录边的数目
    bool vis=false;
    Init();
    if(n==-1&&m==-1) break;

    if(n==0&&m==0) {printf("Yes\n");continue;}

    Union(n,m);sum++;
    s.insert(n);s.insert(m);

    while(1){
        scanf("%d%d",&a,&b);
        if(!(a+b)) break;
        s.insert(a);s.insert(b);
        sum++;
        if(Union(a,b)) vis=true;
    }

    if(s.size()==sum+1&&!vis) printf("Yes\n");
    else printf("No\n");

    }
 return 0;
}

 如果不用set容器用最暴力的方法也可以

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<cmath>
#include<queue>
#include<set>
using namespace std;
typedef long long ll;
#define rep(i,a,b) for(int i=a;i<=b;i++)
const int N=1e6+7;
int m,n;
int f[N];
bool book[N];
int a,b;
void Init(){
    memset(book,0,sizeof book);
    rep(i,0,100000) f[i]=i;
}
int Findf(int v){
    if(f[v]==v) return f[v];
    else{
        f[v]=Findf(f[v]);
        return f[v];
    }
}
bool Union(int a,int b){
    int t1=Findf(a);
    int t2=Findf(b);
    if(t1!=t2)
        {f[t2]=t1;return false;}
    return true;
}

int main()
{
    #ifndef ONLINE_JUDGE
    freopen("in.txt","r",stdin);
    #endif // ONLINE_JUDGE
    while(scanf("%d%d",&n,&m)!=EOF){
    bool vis=false;
    Init();
    if(n==-1&&m==-1) break;
    if(n==0&&m==0) {printf("Yes\n");continue;}

    Union(n,m);
    book[n]=book[m]=1;
    while(1){
        scanf("%d%d",&a,&b);
        if(!(a+b)) break;
        book[a]=book[b]=1;
        if(Union(a,b)) vis=true;
    }
    int sum=0;
    rep(i,1,100000){
        if(book[i]&&f[i]==i) sum++;
    }
    if(!vis&&sum<=1) printf("Yes\n");
    else printf("No\n");

    }
 return 0;
}