【计蒜客】灌溉

到了旱季农业生产的灌溉就成了一个大问题。为了保证灌溉的顺利,某县政府决定投资为各个村之间建立灌溉管道。

输入第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;
	}
	
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值