POJ-1258 Agri-Net (Prim)

題目鏈接：http://poj.org/problem?id=1258

雷同與 poj1789 求最小生成樹的路徑

代碼：

#include<cstdio>
#include<cstring>
#define N 105
#define inf 0x3f3f3f3f
#include<algorithm>
using namespace std;
int map[N][N],dis[N];
int n;
int Prim() {
	int ans=0,minn;
	dis[0]=0;
	for(int i=1; i<n; i++)
		dis[i]=map[0][i];
	for(int i=1; i<n; i++) {
		minn=inf;
		int j,k;
		for(j=0; j<n; j++) {
			if(dis[j]!=0&&dis[j]<minn) {
				minn=dis[j];
				k=j;
			}
		}
		ans+=minn;
		for(j=0; j<n; j++)
			dis[j]=min(dis[j],map[k][j]);
	}
	return ans;
}

int main() {
	while(~scanf("%d",&n)) {
		memset(map,inf,sizeof(map));
		for(int i=0; i<n; i++)
			for(int j=0; j<n; j++)
				scanf("%d",&map[i][j]);
		printf("%d\n",Prim());
	}
}