POJ 2421

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

#include
#include
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
{
min=0x7fffffff;
for(j=1;j<=n;j++)
if(!u[j]&&clost[j]
{
min=clost[j];
now=j;
}
sum+=min;
u[now]=1;
for(j=1;j<=n;j++)
if(!u[j]&&map[now][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
{
scanf("%d%d",&x,&y);
map[x][y]=map[y][x]=0;
}
printf("%d\n",prim());
}
return 0;
}

你可能感兴趣的:(POJ)