最小生成树

大体题意：河流淹没一部分城市，在剩余的城市里有一些连接有一些不链接，要求求出把不连接的城市链接的最少费用。

第一行输入测试组数，第二行输入城市个数n，可选择连接的路线条数m，已经联通的路线条数k；接下来m行表示每条路的起止点和费用；接下来k行表示链接的路线；

思路：用prim最小生成树方法，建立map数组。

感想：学会了的话其实不难，主要是算法没什么思路要求，但主要问题就是没学会

AC 代码：

#include<stdio.h>

#include<string.h>

#define MAX 1000000

int map[510][510];

void prim(int map[][510],int n)

{

int k[510],v[510],i,j;

int min,sum=0,flag,count=1;

memset(k,0,sizeof(k));

memset(v,0,sizeof(v));

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

k[i]=map[1][i];

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

{

min=MAX;

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

{

if(min>k[j]&&v[j]==0)

{

min=k[j];

flag=j;

}

}

if(min!=MAX)

count++;

sum+=min;

v[flag]=1;

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

if(v[j]==0&&k[j]>map[flag][j]&&flag!=j)

k[j]=map[flag][j];

}

if(count==n)

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

else

printf("-1\n");

}

int main()

{

int T,n,m,k,i,j,o,p,q,r,s,l;

int b[100];

scanf("%d",&T);

while(T--)

{

scanf("%d%d%d",&n,&m,&k);

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

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

{

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

else

map[i][j]=MAX;

}

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

{

scanf("%d%d%d",&p,&q,&r);

if(r<map[p][q])

{

map[p][q]=map[q][p]=r;

}

}

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

{

scanf("%d",&s);

for(j=0;j<s;j++)

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

for(o=0;o<s;o++)

for(l=o+1;l<s;l++)

{

map[b[o]][b[l]]=map[b[l]][b[o]]=0;

}

}

prim(map,n);

}

return 0;

}