POJ 2421

  求最小生成树，有点不一样的是有的边已经建立好了，只要将这些建立好的边的权重设为0即可。

#include<stdio.h>

#include<string>

int n,q;

int map[101][101],u[101];

int prim()

{

memset(u,0,sizeof(u));

int i,j,sum=0,min,now;

int clost[101];

for(i=1;i<=n;i++)

clost[i]=map[1][i];

u[1]=1;

for(i=1;i<n;i++)

{

min=0x7fffffff;

for(j=1;j<=n;j++)

if(!u[j]&&clost[j]<min)

{

min=clost[j];

now=j;

}

sum+=min;

u[now]=1;

for(j=1;j<=n;j++)

if(!u[j]&&map[now][j]<clost[j])

clost[j]=map[now][j];

}

return sum;

}

int main()

{

int i,j,x,y;

while(scanf("%d",&n)!=EOF)

{

for(i=1;i<=n;i++)

for(j=1;j<=n;j++)

{

scanf("%d",&map[i][j]);

if(map[i][j]==0)

map[i][j]=0x7fffffff;

map[j][i]=map[i][j];

}

scanf("%d",&q);

for(i=0;i<q;i++)

{

scanf("%d%d",&x,&y);

map[x][y]=map[y][x]=0;

}

printf("%d\n",prim());

}

return 0;

}