poj2421 Constructing Roads

来源：互联网发布：c专家编程pdf高清下载编辑：程序博客网时间：2024/05/16 07:57

可以将已经建好的路初始化为路程为0（这样在进行prim算法求最小生成树时已经建好的路会被首先加入进去，并且不会对我们所要求的还需建设的路径长度有贡献），那么为了避免混淆，到达自身的路程就不能再做初始化为0处理了，这里全都初始化为无穷大，剩下的就是朴素的prim算法了。

#include<iostream>#include<cstring>#include<cstdio>#define INF 0x3f3f3f3fint 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;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;for(j=0;j<n;j++)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;}