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

布线问题

时间限制:1000 ms  |  内存限制:65535 KB
难度:4
描述
南阳理工学院要进行用电线路改造,现在校长要求设计师设计出一种布线方式,该布线方式需要满足以下条件:
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

上一篇:专属博客,你值得拥有(Windows 版本)


下一篇:公司java项目开发规范