NYOJ-ACM-畅通工程

题目：

畅通工程

Time Limit: 4000/2000 MS (Java/Others)    Memory Limit: 65536/32768 K (Java/Others)
Total Submission(s): 62946    Accepted Submission(s): 33683

Problem Description

某省调查城镇交通状况，得到现有城镇道路统计表，表中列出了每条道路直接连通的城镇。省政府“畅通工程”的目标是使全省任何两个城镇间都可以实现交通（但不一定有直接的道路相连，只要互相间接通过道路可达即可）。问最少还需要建设多少条道路？ 

 

Input

测试输入包含若干测试用例。每个测试用例的第1行给出两个正整数，分别是城镇数目N ( < 1000 )和道路数目M；随后的M行对应M条道路，每行给出一对正整数，分别是该条道路直接连通的两个城镇的编号。为简单起见，城镇从1到N编号。 
注意:两个城市之间可以有多条道路相通,也就是说
3 3
1 2
1 2
2 1
这种输入也是合法的
当N为0时，输入结束，该用例不被处理。 

 

Output

对每个测试用例，在1行里输出最少还需要建设的道路数目。 

 

Sample Input

4 21 34 33 31 21 32 35 21 23 5999 00

 

Sample Output

102998

Hint

Hint

Huge input, scanf is recommended.

思路：

这道题用到了并查集，用find()函数来查找祖先节点。用join()连通已知的两个节点。用bool t()来存放和判断有几个集合。具体的有代码注释。有什么问题欢迎一起讨论。

优化后代码：

#include<iostream>
#include<algorithm>
#include<cstring> 
#include<cstdio>
using namespace std;
int pre[1010];
bool t[1010];
int find(int x){
	int r = x;
	while(pre[r] != r){//寻找自己的祖先
		r = pre[r];
	}
	int i = x,j;
	while(pre[i] != r){//路径压缩，减少时间复杂度
		j = pre[i];
		pre[i] = r;
		i = j;
	}
	return r;
}
void join(int a,int b){
	int fx = find(a);
	int fy = find(b);
	if(fx != fy){//如果不连通，就变成连通
		pre[fx] = fy;
	}
}
int main(){

	int n,m,p,q;
	//freopen("data.txt","r",stdin);
	while(scanf("%d",&n)!=EOF){
		int co = 0;
		if(n == 0){
			 break;
		}else {
			scanf("%d",&m);
			if(m == 0){
				printf("%d\n",n-1);
			}else {
				for(int i=1;i<=n;i++){//初始化pre[i]
					pre[i] = i;
				}
				for(int i=1;i<=m;i++){
					scanf("%d%d",&p,&q);
					join(p,q);	
				}
				for(int i=1;i<=n;i++){
					if(pre[i] == i){
						co ++;
					}
				} 
				printf("%d\n",co-1);//连通
			}
		}
	}
	return 0;
}

代码：

#include<iostream>
#include<algorithm>
#include<cstring> 
#include<cstdio>
using namespace std;
int pre[1010];
bool t[1010];
int find(int x){
	int r = x;
	while(pre[r] != r){//寻找自己的祖先
		r = pre[r];
	}
	int i = x,j;
	while(pre[i] != r){//路径压缩，减少时间复杂度
		j = pre[i];
		pre[i] = r;
		i = j;
	}
	return r;
}
void join(int a,int b){
	int fx = find(a);
	int fy = find(b);
	if(fx != fy){//如果不连通，就变成连通
		pre[fx] = fy;
	}
}
int main(){

	int n,m,p,q;
	while(scanf("%d",&n)!=EOF){
		if(n == 0){
			 break;
		}else {
			scanf("%d",&m);
			if(m == 0){
				printf("%d\n",n-1);
			}else {
				for(int i=1;i<=n;i++){//初始化pre[i]
					pre[i] = i;
				}
				for(int i=1;i<=m;i++){
					scanf("%d%d",&p,&q);
					join(p,q);	
				}
				memset(t,0,sizeof(t));//初始化
				int co = 0;
				for(int i=1;i<=n;i++){
					t[find(i)] = 1;//判断有几个连通的。例如输入4 2 1 3 4 3。pre[1]=3,pre[2]=2,pre[3]=3,pre[4]=3。
				}					//就有t[1]=0,t[2]=1,t[3]=1,t[4]=0。就有两个集合即{1,3,4}和{2}。所以只要一条路径就可以
									//如果输入999 0 就会有999个集合，但是要连通的话，并不需要999条路，998条就完成了工作
				for(int i=1;i<=n;i++){
					if(t[i] == true){
						co ++;
					}
				}
				printf("%d\n",co-1);//连通
			}
		}
	}
	return 0;
}