畅通工程_hdu_1863(并查集+最小生成树).java

畅通工程

Time Limit: 1000/1000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 13018    Accepted Submission(s): 5321

Problem Description

省政府“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通（但不一定有直接的公路相连，只要能间接通过公路可达即可）。经过调查评估，得到的统计表中列出了有可能建设公路的若干条道路的成本。现请你编写程序，计算出全省畅通需要的最低成本。

 

Input

测试输入包含若干测试用例。每个测试用例的第1行给出评估的道路条数 N、村庄数目M ( < 100 )；随后的 N 
行对应村庄间道路的成本，每行给出一对正整数，分别是两个村庄的编号，以及此两村庄间道路的成本（也是正整数）。为简单起见，村庄从1到M编号。当N为0时，全部输入结束，相应的结果不要输出。

 

Output

对每个测试用例，在1行里输出全省畅通需要的最低成本。若统计数据不足以保证畅通，则输出“?”。

 

Sample Input

  
  

   
   3 3
1 2 1
1 3 2
2 3 4
1 3
2 3 2
0 100
  
  

 

Sample Output

  
  

   
   3
?
  
  

 

Source

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

 

import java.util.Arrays;
import java.util.Scanner;

public class Main{//听说可以用dijkstra，floyd，但我感觉还是用：并查集+最小生成树方便
	static int rank[],p[];
	static Map map[];
	public static void main(String[] args) {
		Scanner input=new Scanner(System.in);
		while(true){
			int n=input.nextInt();
			int m=input.nextInt();
			if(n==0)
				break;
			Init(m,n);
			for(int i=0;i<n;i++){
				int x=input.nextInt();
				int y=input.nextInt();
				int cost=input.nextInt();
				map[i]=new Map(x,y,cost);
			}
			Arrays.sort(map);
			long sum=0;
			for(int i=0;i<n;i++){//排序求最小生成树
				if(merge(map[i].x,map[i].y)){//并查集，记录是否被联通
					sum+=map[i].cost;
				}
			}
			int num=0;
			for(int i=1;i<=m;i++){
				if(p[i]==i)
					num++;
			}
			if(num>1)//是否是连通图
				System.out.println("?");
			else
				System.out.println(sum);
		}
	}
	private static boolean merge(int x, int y) {
		int X=find(x);
		int Y=find(y);
		if(X==Y)
			return false;
		if(rank[X]<rank[Y]){//树的深度优化
			p[X]=Y;
		}else{
			p[Y]=X;
			if(rank[X]==rank[Y])
				rank[Y]++;
		}
		return true;
	}
	private static int find(int x) {//查找根节点
		while(x!=p[x])
			x=p[x];
		return x;
	}
	private static void Init(int m,int n) {//初始化
		map=new Map[n];
		rank=new int[m+1];
		p=new int[m+1];
		for(int i=1;i<=m;i++)
			p[i]=i;
		Arrays.fill(rank, 1);
		
	}
}
class Map implements Comparable<Map>{
	int x,y;
	int cost;
	public Map(int x, int y, int cost) {//属性赋值
		this.x = x;
		this.y = y;
		this.cost = cost;
	}
	@Override
	public int compareTo(Map o) {//重写排序方法
		
		return this.cost-o.cost;
	}
}