佈線問題_ny_38(並查集+最小生成樹).java

描述

南陽理工學院要進行用電線路改造，現在校長要求設計師設計出一種佈線方式，該佈線方式需要滿足以下條件：
1、把所有的樓都供上電。
2、所用電線花費最少

輸入

第一行是一個整數n表示有n組測試資料。(n<5)
每組測試資料的第一行是兩個整數v,e.
v表示學校裡樓的總個數(v<=500)
隨後的e行裡，每行有三個整數a,b,c表示a與b之間如果建鋪設線路花費為c(c<=100)。（哪兩棟樓間如果沒有指明花費，則表示這兩棟樓直接連通需要費用太大或者不可能連通）
隨後的1行裡，有v個整數,其中第i個數表示從第i號樓接線到外界供電設施所需要的費用。( 0<e<v*(v-1)/2 )
（樓的編號從1開始），由於安全問題，只能選擇一個樓連線到外界供電裝置。
資料保證至少存在一種方案滿足要求。

輸出

每組測試資料輸出一個正整數,表示鋪設滿足校長要求的線路的最小花費。

樣例輸入

1
4 6
1 2 10
2 3 10
3 1 10
1 4 1
2 4 1
3 4 1
1 3 5 6

樣例輸出

4

來源

[張雲聰]原創

上傳者

張雲聰

 

import java.util.Arrays;
import java.util.Comparator;
import java.util.Scanner;
public class Main {//最小生成樹+並查集
	private static int d[];
	private static Point point[];
	public static void main(String[] args) {
		Scanner input=new Scanner(System.in);
		int N=input.nextInt();
		while(N-->0){
			int v=input.nextInt();
			int e=input.nextInt();
			d=new int[v+1];
			for(int i=1;i<=v;i++)
				d[i]=i;
			point=new  Point[e];
			long sum=Long.MAX_VALUE;
			for(int i=0;i<e;i++){
				int a=input.nextInt();
				int b=input.nextInt();
				int c=input.nextInt();
				point[i]=new Point(a,b,c);
			}
			for(int i=0;i<v;i++){
				int a=input.nextInt();
				if(sum>a)
					sum=a;
			}
			Arrays.sort(point,0,e,new Comparator<Point>(){
				@Override
				public int compare(Point o1, Point o2) {
					if(o1.getP()>o2.getP())
						return 1;
					else if(o1.getP()<o2.getP()){
						return -1;
					}else
						return 0;
				}
			});
			for(int i=0;i<e;i++){
				if(hebing(point[i].getX(),point[i].getY())){
					sum+=point[i].getP();
				}
			}
			System.out.println(sum);
		}
	}
	private static boolean hebing(int x, int y) {
		int X=find(x);
		int Y=find(y);
		if(X==Y){
			return false;
		}
		d[X]=Y;
		return true;
	}
	private static int find(int x) {
		while(x!=d[x])
			x=d[x];
		return x;
	}
}
class Point{
	int x,y,p;
	
	public Point(int x, int y, int p) {
		this.x = x;
		this.y = y;
		this.p = p;
	}
	public int getX() {
		return x;
	}
	public int getY() {
		return y;
	}
	public int getP() {
		return p;
	}
}