PTA：文件传输 (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.

结尾无空行

作者：陈越；单位：浙江大学；代码长度限制：16 KB；时间限制：150 ms；内存限制：64 MB

这道题是个并查集的模板题。在此基础上题目要求判断网络是否完全相连，即为集合的个数。集合个数为1说明全部相连了，否则输出集合个数。

code:

#include<iostream>
using namespace std;
int s[10005],h[10005]={0}; //s为并查集，h为高度
int find_set(int x) //查找
{
	if(x!=s[x]) return find_set(s[x]);
	return s[x];
}
void union_set(int x,int y)
{
	x=find_set(x);
	y=find_set(y);
	if(h[x]==h[y]){ //高度相同并到左边
		s[y]=x;
		h[x]++;
	}
	else{ //矮树并高树
		if(h[x]<h[y]) s[x]=y;
		else s[y]=x;
	}
}
int main()
{
	ios::sync_with_stdio(0);
	int n;
	cin>>n;
	for(int i=1;i<=n;i++) s[i]=i;//初始化
	while(1){
		char ch;
		int x,y;
		cin>>ch;
		if(ch=='S') break;
		cin>>x>>y;
		if(ch=='I')
		union_set(x,y);
		else{
			if(find_set(x)==find_set(y)) cout<<"yes"; //如果在同一集合说明连接
			else cout<<"no";
			cout<<endl;
		} 
	}
	int k=0;
	for(int i=1;i<=n;i++) if(find_set(i)==i) k++; //统计集合个数
	if(k==1) cout<<"The network is connected.";
	else cout<<"There are "<<k<<" components.";
	return 0;
}

结果：

 关于并查集，推荐知乎文章链接：

算法学习笔记(1) : 并查集 - Pecco的文章 - 知乎

https://zhuanlan.zhihu.com/p/93647900