1350-最短网络(agrinet)（最小生成树）---信息学奥赛一本通

【输入样例】

4
0 4 9 21
4 0 8 17
9 8 0 16
21 17 16 0

【输出样例】

28

 

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=110;
int n,w[N][N],dis[N],vis[N];
int prim(){
	int res=0;
	memset(dis,0x3f,sizeof dis);
	dis[1]=0;
	for(int i=1;i<=n;i++){
		int t=-1;
		for(int j=1;j<=n;j++){
			if(!vis[j]&&(t==-1||dis[t]>dis[j])) t=j;
		}	
		res+=dis[t];
		vis[t]=1;
		for(int j=1;j<=n;j++) dis[j]=min(dis[j],w[t][j]);
	}
	return res;
}
int main(){
	scanf("%d",&n);
	for(int i=1;i<=n;i++)
		for(int j=1;j<=n;j++)
			scanf("%d",&w[i][j]);
	printf("%d",prim());
	return 0;
}