2019 GDUT 新生专题Ⅲ I题

I - 还是畅通工程

题目链接
题目大意：计算能连通所有村庄的最短路线。
思路：直接套最小生成树模板prim或者Kruskal。下面代码用了prim。
代码

#include <cstdio>
#include <algorithm>
using namespace std;
const int MAX=105;
const int inf=999999999;
int mincost[MAX],cost[MAX][MAX],n;
bool used[MAX];
int prim(){
	mincost[1]=0;
	int res=0;
	while(true){
		int v=-1;
		for(int i=1;i<=n;i++){
			if(!used[i]&&(v==-1||mincost[i]<mincost[v])) v=i;
		}
		if(v==-1) break;
		used[v]=true;
		res+=mincost[v];
		for(int i=1;i<=n;i++)
		mincost[i]=min(mincost[i],cost[v][i]);
	}
	return res;
}
int main(){
	int u,v,d;
	while(scanf("%d",&n)==1&&n){
		int m=(n-1)*n/2;
		fill(cost[0],cost[0]+MAX*MAX,inf);
		fill(mincost,mincost+MAX,inf);
		fill(used,used+MAX,false);
		for(int i=1;i<=n;i++) cost[i][i]=0;
		for(int i=0;i<m;i++){
			scanf("%d%d%d",&u,&v,&d);
			cost[u][v]=cost[v][u]=d;
		}
		int res=prim();
		printf("%d\n",res);
	}
	return 0;
}