最小生成树 , Prim算法
注意:
1.若第一次添加边之后min等于max,证明此图不连通,则输出-1 ;不然继续。
2.将已经联通的城市的map[][]置为0 ,不在选择范围内 。
AC代码:
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <math.h>
int map[505][505], v[505], low[505], t[105];
int n, sum, min, flag;
const int max = 0x3fffffff;
int MinTree(int n)
{
int i,j;
memset(low, 0, sizeof(low));
memset(v, 0, sizeof(v));
v[1] = 1; //从点1开始贪心
sum = 0;
for(i=1; i<=n; i++)
{
low[i] = map[1][i]; //从点1到点i的边的权值加入数组low[]
}
for(i=2; i<=n; i++) //从点2开始更新边
{
min = max;
flag = -1;
for(j=2; j<=n; j++)
{
if(!v[j] && low[j]<min)
{
min = low[j];
flag = j;
}
}
if(min == max) return -1; //此图不连通
sum += min;
v[flag] = 1;
for(j=1; j<=n; j++) //更新边
{
if(!v[j] && map[flag][j]<low[j])
{
low[j] = map[flag][j];
}
}
}
return sum;
}
int main()
{
int T, m, k, p, q, c, tt;
int i, j, l, ans;
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, &c);
if(c < map[p][q])
{
map[p][q] = c;
map[q][p] = c;
}
}
for(i=0; i<k; i++)
{
scanf("%d", &tt);
for(j=0; j<tt; j++)
{
scanf("%d", &t[j]);
}
for(j=0; j<tt-1; j++) //已连接的边不在选择范围内
{
for(l=j+1; l<tt; l++)
{
map[t[j]][t[l]] = 0;
map[t[l]][t[j]] = 0;
}
}
}
//printf("%d\n\n", max);
ans = MinTree(n);
printf("%d\n", ans);
}
return 0;
}