暢通工程_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

HintHint
Huge input, scanf is recommended.

Source
 
 

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;
	}
}