版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/qq_35131940/article/details/79123819
到了旱季农业生产的灌溉就成了一个大问题。为了保证灌溉的顺利,某县政府决定投资为各个村之间建立灌溉管道。
输入第1行包括一个整数N,表示某县的村庄的数量。(3≤N≤100),第2行-结尾为一个N×N的矩阵,表示每个村庄之间的距离。虽然在理论上,他们是N行,每行由N个用空格分隔的数组成,实际上,他们限制在80个字符,因此,某些行会紧接着另一些行。当然,对角线将会是0,因为不会有线路从第i个村到它本身(任何两个村之间的距离都不大于100000)。
输出只有一个,为修建灌溉管道将所有村庄相连在一个灌溉系统里所需的最小管道长度。
样例输入
4 0 4 9 21 4 0 8 17 9 8 0 16 21 17 16 0
样例输出
28
想法就是连通图+最小优先树。
import java.util.PriorityQueue;
import java.util.Queue;
import java.util.Scanner;
public class Main {
int con[];
public static void main(String args[]){
Main test=new Main();
test.start();
}
public void start(){
Scanner scan=new Scanner(System.in);
int num=scan.nextInt();
con=new int[num];
for(int i=0;i<num;i++){
con[i]=i;
}
int value[]=new int[num];
Queue<Edge> queue=new PriorityQueue<Edge>();
for(int i=0;i<num;i++){
for(int j=0;j<num;j++){
int w=scan.nextInt();
if(w!=0)
queue.offer(new Edge(i,j,w));
}
}
int min=0;
while(queue.size()!=0){
Edge a=queue.poll();
int i1=a.s;
int j1=a.e;
if(find(i1)==find(j1))
continue;
min+=a.w;
union(i1,j1);
}
System.out.println(min);
}
public void union(int i,int j){
int i1=find(i);
int j1=find(j);
if(i1==j1) return;
con[i1]=j1;
}
public int find(int i){
if(con[i]!=i) return find(con[i]);
return i;
}
}
class Edge implements Comparable<Edge>{
int s;
int e;
int w;
public Edge(int s, int e, int w) {
super();
this.s = s;
this.e = e;
this.w = w;
}
@Override
public int compareTo(Edge o) {
if(w>((Edge)o).w)
return 1;
return -1;
}
}