poj1258 Agri-Net

最小生成树模板题，采用prim算法

#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
int ma[101][101];
int cost[101]; 
int n;
int prim(int s){
	int i,j;
	for(i=0;i<n;i++)
		cost[i]=ma[s][i];
	int sum=0;
	int mini;
	for(i=0;i<n-1;i++){
		mini=0x3f3f3f3f;
		for(j=0;j<n;j++)
			if(cost[j]<mini&&cost[j]!=0){
				s=j;
				mini=cost[j];
			}
			sum+=mini;
			cost[s]=0;
			for(j=0;j<n;j++)
				if(cost[j]>ma[s][j]&&ma[s][j]!=0)
					cost[j]=ma[s][j];
	}
	return sum;
}
int main(){
	int i,j;
	while(scanf("%d",&n)!=EOF){
		for(i=0;i<n;i++)
			for(j=0;j<n;j++)
				scanf("%d",&ma[i][j]);
		printf("%d\n",prim(0));
	}
	return 0;
}