【并查集】问题 A: 通信系统

题目描述

某市计划建设一个通信系统。按照规划，这个系统包含若干端点，这些端点由通信线缆链接。消息可以在任何一个端点产生，并且只能通过线缆传送。每个端点接收消息后会将消息传送到与其相连的端点，除了那个消息发送过来的端点。如果某个端点是产生消息的端点，那么消息将被传送到与其相连的每一个端点。

为了提高传送效率和节约资源，要求当消息在某个端点生成后，其余各个端点均能接收到消息，并且每个端点均不会重复收到消息。

现给你通信系统的描述，你能判断此系统是否符合以上要求吗？

输入

输入包含多组测试数据。每两组输入数据之间由空行分隔。

每组输入首先包含2个整数N和M，N（1<=N<=1000）表示端点个数，M（0<=M<=N*(N-1)/2）表示通信线路个数。

接下来M行每行输入2个整数A和B（1<=A，B<=N），表示端点A和B由一条通信线缆相连。两个端点之间至多由一条线缆直接相连，并且没有将某个端点与其自己相连的线缆。

当N和M都为0时，输入结束。

输出

对于每组输入，如果所给的系统描述符合题目要求，则输出Yes，否则输出No。

样例输入 复制

4 3
1 2
2 3
3 4

3 1
2 3

0 0

样例输出 复制

Yes
No

分析：

题目要求判断若干端点是否连接，且各端点之间只有一条通路。可以使用并查集的方式判断，每次读入两个直接连通的端点，判断并查集中两者是否是同一集合，若是则表明有这两个端点会有多条通路，若不是则将两个端点并入一个集合中。当集合并入完毕，遍历一次全部端点，查看其是否都为同一集合。

要点：

1. 在两个端点所在集合的并操作中，若两个不在同一集合中，是要将一个集合并入另一个集合，而不是将该端点并入另一个集合，因此是该集合的根结点的并入；

2. 注意题目条件，是在N和M都为0时才结束，因此只要有一个不为0即继续。

#include <iostream>

using namespace std;

int father[1010];
bool flag;

void Initial(int a){
    for(int i = 1; i <= a; i++){
        father[i] = i;
    }
}
int findFather(int x){
    int a = x;
    while(father[x] != x){
        x = father[x];
    }

    int z = a;
    while(father[a] != a){
        z = a;
        a = father[a];
        father[z] = x;
    }
    return x;
}
void Union(int a, int b){
    int fa = findFather(a);
    int fb = findFather(b);
    if(fa != fb){
        father[fb] = fa;
    }else{
        flag = false;
    }
    return;
}
bool isSet(int a){
    int f = findFather(1);
    for(int i = 2; i <= a; i++){
        if(f != findFather(i)) return false;
    }
    return true;
}
int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);

    int N, M;
    while(cin >> N >> M && (M != 0 || N != 0)){
        Initial(N);
        flag = true;
        for(int i = 0; i < M; i++){
            int a, b;
            cin >> a >> b;
            Union(a, b);
        }
        if(flag && isSet(N)){
            cout << "Yes\n";
        }else{
            cout << "No\n";
        }
    }
    return 0;
}