畅通工程_hdu_1232(并查集)

畅通工程

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

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 2 1 3 4 3 3 3 1 2 1 3 2 3 5 2 1 2 3 5 999 0 0

 

Sample Output

1 0 2 998

Hint

Hint

Huge input, scanf is recommended.

 

Source

浙大计算机研究生复试上机考试-2005年

 

import java.util.Scanner;
public class Main {
	private static int a[];

	public static void main(String[] args) {
		Scanner input=new Scanner(System.in);
		while(true){
			int n=input.nextInt();
			if(n==0)
				break;
			a=new int[n+1];
			for(int i=1;i<=n;i++){
				a[i]=i;
			}
			int m=input.nextInt();
			for(int i=0;i<m;i++){
				int x=input.nextInt();
				int y=input.nextInt();
				hebing(x,y);
			}
			int sum=0;
			for(int i=1;i<=n;i++){
				if(a[i]==i) sum++;
			}
			System.out.println(sum-1);
		}

	}
	private static void hebing(int x, int y) {
		int x1=find(x);
		int y1=find(y);
		a[x1]=y1;
	}
	private static int find(int x) {
		int r=x;
		while(a[r]!=r)
			r=a[r];
		return r;
	}
}

优化后

import java.util.Scanner;
public class Main {//优化后代码，二叉优化
	private static int a[];
	public static void main(String[] args) {
		Scanner input=new Scanner(System.in);
		while(true){
			int n=input.nextInt();
			if(n==0)
				break;
			a=new int[n+1];
			for(int i=1;i<=n;i++){
				a[i]=i;
			}
			int m=input.nextInt();
			for(int i=0;i<m;i++){
				int x=input.nextInt();
				int y=input.nextInt();
				hebing(x,y);
			}
			int sum=0;
			for(int i=1;i<=n;i++){
				if(a[i]==i) sum++;
			}
			System.out.println(sum-1);
		}

	}
	private static void hebing(int x, int y) {
		int x1[]=new int[2];
		int y1[]=new int[2];
		x1=find(x);
		y1=find(y);
		if(x1[0]!=y1[0])
			if(x1[1]<y1[1]){
				a[x1[0]]=y1[0];
				a[x]=y1[0];
			}
			else{
				a[y1[0]]=x1[0];
				a[y]=x1[0];
			}
	}
	private static int[] find(int x) {
		int b[]=new int[2];
		b[0]=x;b[1]=0;
		while(a[b[0]]!=b[0]){
			b[0]=a[b[0]];
			b[1]++;
		}
		return b;
	}
}