KY126 畅通工程（用Java实现）

描述

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

输入描述：

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

输出描述：

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

示例1

输入：

4 2
1 3
4 3
3 3
1 2
1 3
2 3
5 2
1 2
3 5
999 0
0

复制输出：

1
0
2
998

代码如下：

import java.util.Scanner;
import java.util.Set;
import java.util.TreeSet;
/*
*    将城镇抽象为顶点，将有道路联通的城镇划分到同一个集合中，将问题转化成求集合个数的问题
*/
public class ChangTongEngine {

	public static void main(String[] args) {
		
		Scanner scanner = new Scanner(System.in);
		while (scanner.hasNextInt()) {
			int n = scanner.nextInt();	//n表示城镇数量
			if (n == 0) {
				break;
			}
			int m = scanner.nextInt();	//m表示道路数目
			
			//初始化一个并查集
			UnionFind uf = new UnionFind(n+1);
			
			for (int i = 0; i < m; i++) {
				//m1和m2表示两个城镇编号
				int m1 = scanner.nextInt();
				int m2 = scanner.nextInt();
				//合并已经建成的城市之间的道路
				uf.union(m1, m2);
				
			}
			//TreeSet中元素是有序且不重复的
			TreeSet<Integer> ts = new TreeSet<Integer>();
			
			for (int i = 1; i <= n; i++) {
				ts.add(uf.find(i));
			}
			
			//n个节点至少需要n-1条边来连接
			System.out.println(ts.size() - 1);
			
		}
	}
}

//并查集类
class UnionFind{
	
	int[] parent;	//记录结点的父亲
	int[] height;	//记录树的高度
	
	//有参数构造方法
	//初始化并查集
	public UnionFind(int n) {
		parent = new int[n];
		height = new int[n];
		for (int i = 0; i < parent.length; i++) {
			
			parent[i] = i;	//初始时，每个节点的父亲节点都是自己
			height[i] = 0;	//初始时，所有树高均为1
		}
	}
	
	//查操作，找到元素x的根结点
	public int find(int x) {
		if (x != parent[x]) {
			//压缩路径：先找到根结点，再将查找路径上所有结点都挂到根结点下
			parent[x] = find(parent[x]);
		}
		return parent[x];
	}
	
	//合并操作
	public void union(int x,int y) {
		
		int rootX = find(x);
		int rootY = find(y);
		
		//如果两颗树的根结点相同时，说明两个节点在一个集合中
		if (rootX == rootY) {
			return;
		}else {
			//按高度合并
			if (height[rootX] > height[rootY]) {
				parent[rootY] = rootX;
			}else if (height[rootX] < height[rootY]) {
				parent[rootX] = rootY;
			}else{
				parent[rootY] = rootX;
				//如果高度相同则合并后高度加1
				height[rootX]++;
			}
		}
		
		
		for (int i = 0; i < parent.length; i++) {
			if (parent[i] == rootX) {	
				//遍历并查集中的结点，如果并查集中结点的祖先是x的祖先，则全部设置成y的祖先
				parent[i] = rootY;
			}
		}
		
	}
}