布线问题_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;
	}
}

布线问题_ny_38(并查集+最小生成树).java,布布扣,bubuko.com

布线问题_ny_38(并查集+最小生成树).java

原文：http://blog.csdn.net/zhangyiacm/article/details/21252421