POJ2421 Constructing Roads

最小生成树，采用prim算法，对于已经建好的路，初始化两点之间距离为零。对于到达自身的距离，题目给的输入是0，为避免和已经建好的路产生矛盾，初始化为无穷大。剩下的便是最朴素的prim算法解决。

#include<iostream>
#include<cstring>
#include<cstdio>
#define INF 0x3f3f3f3f
int ma[105][105];
int cost[105];
int n,m;
int prim(int s){
	int i,j;
	for(i=0;i<n;i++)
		cost[i]=ma[s][i];
	cost[s]=-1;   //首先选入点s，并置-1避免再次选取
	int sum=0;
	int mini;
	for(i=0;i<n-1;i++){
		mini=INF;
		for(j=0;j<n;j++)
			if(cost[j]<mini&&cost[j]!=-1){   //选取距离最小的&&未被选取过的
				s=j;
				mini=cost[j];
			}
		sum+=mini;
		cost[s]=-1;   //置-1避免重复选取
		for(j=0;j<n;j++)   //更新cost[]
			if(cost[j]>ma[s][j])
				cost[j]=ma[s][j];
	}
	return sum;
}
int main(){
	int i,j;
	int a,b;
	while(scanf("%d",&n)!=EOF){
		for(i=0;i<n;i++)
			for(j=0;j<n;j++){
				scanf("%d",&ma[i][j]);
				if(i==j)
					ma[i][i]=INF;   //对于到达自身的初始化为无穷大
			}
			scanf("%d",&m);
			for(i=0;i<m;i++){
				scanf("%d%d",&a,&b);
				--a;
				--b;
				ma[a][b]=ma[b][a]=0;   //无向图，对称位置同样初始化为零 
			}
			printf("%d\n",prim(0));
	}
	return 0;
}