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

时间:2022-09-10 21:35:10

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

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