L2-2 文件传输 (25 分)——寒假训练测试题目集

当两台计算机双向连通的时候，文件是可以在两台机器间传输的。给定一套计算机网络，请你判断任意两台指定的计算机之间能否传输文件？

输入格式：

首先在第一行给出网络中计算机的总数 N (2≤N≤104)，于是我们假设这些计算机从 1 到 N 编号。随后每行输入按以下格式给出：

I c1 c2  

其中I表示在计算机c1和c2之间加入连线，使它们连通；或者是

C c1 c2    

其中C表示查询计算机c1和c2之间能否传输文件；又或者是

S

这里S表示输入终止。

输出格式：

对每个C开头的查询，如果c1和c2之间可以传输文件，就在一行中输出"yes"，否则输出"no"。当读到终止符时，在一行中输出"The network is connected."如果网络中所有计算机之间都能传输文件；或者输出"There are k components."，其中k是网络中连通集的个数。

输入样例 1：

5
C 3 2
I 3 2
C 1 5
I 4 5
I 2 4
C 3 5
S

输出样例 1：

no
no
yes
There are 2 components.

输入样例 2：

5
C 3 2
I 3 2
C 1 5
I 4 5
I 2 4
C 3 5
I 1 3
C 1 5
S

输出样例 2：

no
no
yes
yes
The network is connected.

AC代码：

思路： 属于是并查集的考察，在每一次询问都要重新查找TA们的祖先是否是同一个，最后统计图的连通分支数进行判断并输出。

#include<bits/stdc++.h>
using namespace std;
int fa[10001];
void init(int n){
    for(int i=1;i<=n;i++){//注意计算机从1开始编号
        fa[i]=i;
    }
}
int find(int a){
    return a==fa[a]?a:fa[a]=find(fa[a]);
}
void merge(int a,int b){
    int p1=find(a);
    int p2=find(b);
    if(p1!=p2){
        fa[p2]=p1;
    }
}
int main(){
    int n;
    cin>>n;
    init(n);
    char op;
    cin>>op;
    while(op!='S'){
        int a,b;
        cin>>a>>b;
        if(op=='C'){
            cout<<(find(a)==find(b)?"yes":"no")<<endl;
        }else {
            merge(a,b);
        }
        cin>>op;
    }
    int cnt=0;
    for(int i=1;i<=n;i++){
        if(fa[i]==i){
            cnt++;
        }
    }
    if(cnt==1){
        cout<<"The network is connected."<<endl;
    }else{
         printf("There are %d components.\n", cnt);
    }
    return 0;
}