布线问题
时间限制: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; } }