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;
}