计蒜客 灌溉 (最小生成树之prim算法)

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

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


生成树之prim算法;

AC代码:

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
#define INF 100000000
const int maxn=100+5;
int g[maxn][maxn];
int vis[maxn];
int d[maxn];
int N;

int prim(){
    int ans=0;
    memset(vis,0,sizeof(vis));
    for(int i=0;i<N;i++)
    d[i]=g[0][i];
    vis[0]=1;
	
	for(int i=0;i<N-1;i++){
		int Min=INF;
		int t; 
		for(int j=0;j<N;j++){
		    if(!vis[j]&&Min>d[j]){
		    	Min=d[j];
		    	t=j;
			}
		}
		vis[t]=1;
		ans+=Min;
		
		for(int k=0;k<N;k++){
			d[k]=min(d[k],g[t][k]);
		}
	}
    return ans;
}

int main(){
	while(scanf("%d",&N)==1){
		for(int i=0;i<N;i++){
			for(int j=0;j<N;j++)
			scanf("%d",&g[i][j]);
		}
		
		int ans = prim();
		printf("%d\n",ans);
	}
	return 0;
} 




  • 1
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

柏油

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值