POJ - 1258 Agri-Net

https://vjudge.net/problem/POJ-1258

模板题，直接prim就行

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
const int N=110;
int n,m;
int g[N][N],dis[N];
bool bk[N];
int prim()
{
	memset(bk,false,sizeof bk);
	memset(dis,0x3f,sizeof dis);
	int ans=0;
	for(int i=0;i<n;i++)
	{
		int t=-1;
		for(int j=1;j<=n;j++)
		{
			if(!bk[j] && (t==-1 || dis[t]>dis[j]))
				t=j;
		}
		bk[t]=1;
		if(i) ans+=dis[t];
		
		for(int j=1;j<=n;j++)
			if(dis[j]>g[t][j])
				dis[j]=g[t][j];
	}
	return ans;
}
int main()
{
	while(~scanf("%d",&n))
	{
		memset(g,0x3f,sizeof g);
		for(int i=1;i<=n;i++)
		{
			for(int j=1;j<=n;j++)
				scanf("%d",&g[i][j]);
		}
		
		printf("%d\n",prim());
	}
	return 0;
}