Hdu 3371 - Connect the Cities

最小生成树 ， 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;
}