佈線問題
- 描述
-
南陽理工學院要進行用電線路改造,現在校長要求設計師設計出一種佈線方式,該佈線方式需要滿足以下條件:
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;
}
}