hud Connect the Cities

注意：后面的有3个城市是指相互连接。

用模板。

 

#include<stdio.h>
#include<string.h>
const int maxn = 510;
#define INF 0xfffffff
int map[maxn][maxn],lowcost[maxn],vis[maxn],n,m,k;

int prim()
{
    int i,j,z,mins,sum=0,yes;
    for(i=1;i<=n;i++) lowcost[i] = map[1][i];
    memset(vis,0,sizeof(vis));
    vis[1]=1;
    for(;;)
    {
        mins=INF;
        yes=-1;
        for(j=1;j<=n;j++)
        {
            if(!vis[j] && lowcost[j] < mins)
            {
                mins = lowcost[j];
                yes=1;
                z=j;
            }
        }
        if(yes==-1) break;
        sum+=mins;
        vis[z] = 1;
        for(j=1;j<=n;j++)
            if(!vis[j] && lowcost[j] > map[z][j]) 
                lowcost[j] = map[z][j];
    }
    for(i=1;i<=n;i++) if(!vis[i]) return -1;
    return sum;
}

int main()
{
    int T,i,j;
    int x,y,z;
    int num;
    int a[maxn];
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d%d%d",&n,&m,&k);
        for(i=1;i<=n;i++)
        {
            for(j=1;j<=n;j++) map[i][j] = map[j][i] = INF;
            map[i][i] = 0;
        }
        for(i=1;i<=m;i++)
        {
            scanf("%d%d%d",&x,&y,&z);
            if(z<map[x][y]) map[x][y] = z;
            if(z<map[y][x]) map[y][x] = z;
        }
        while(k--)
        {
            scanf("%d",&num);
            for(i=1;i<=num;i++) scanf("%d",&a[i]);
            for(i=1;i<num;i++)
                for(j=i+1;j<=num;j++) map[a[i]][a[j]] = map[a[j]][a[i]] = 0;
        }
        printf("%d\n",prim());
    }
    return 0;
}