hdu 3371 Connect the Cities (最小生成树Prim)

题目连接：http://acm.hdu.edu.cn/showproblem.php?pid=3371

题目不难 稍微注意一下 要把已经建好的城市之间的花费定义为0，在用普通Prim算法就可以了；我没有用克鲁斯卡尔算法(Kruskal's algorithm)，因为这题数据比较大，而且要处理大量的数据使它为0，怕超时T^T。。。。。

#include<iostream>
#include<cstdio>
#include<algorithm>

using namespace std;

#define inf 999999

int map[505][505],node[505],n,vis[505];
int dd[505],sum;

int Prim()
{
    int tm=1;
    node[tm]=0;
    //vis[tm]=1;
    for(int k=1;k<=n;k++)
    {
        int minn=inf;
        for(int i=1;i<=n;i++)
        if(!vis[i]&&minn>node[i])
        {
            tm=i;
            minn=node[i];
        }
        vis[tm]=1;
        if(minn==inf)
        return 0;

        sum+=minn;

        for(int i=1;i<=n;i++)
        {
            if(!vis[i]&&node[i]>map[tm][i])
            node[i]=map[tm][i];
        }
    }
    return 1;
}

int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        int m,k;
        scanf("%d%d%d",&n,&m,&k);
        for(int i=1;i<=n;i++)
        {
            node[i]=inf;vis[i]=0;
            for(int j=1;j<=n;j++)
            map[i][j]=inf;
        }
        while(m--)
        {
            int a,b,d;
            scanf("%d%d%d",&a,&b,&d);
            if(map[a][b]>d)
            map[a][b]=map[b][a]=d;
        }
        while(k--)
        {
            int t;
            scanf("%d",&t);
            for(int i=1;i<=t;i++)
            {
                scanf("%d",&dd[i]);
            }
            for(int i=1;i<=t;i++)
            {
                for(int j=i+1;j<=t;j++)
                map[dd[i]][dd[j]]=map[dd[j]][dd[i]]=0;
            }
        }
        sum=0;
        int flag=Prim();
        if(flag==0)
        printf("-1
");
        else
        printf("%d
",sum);
    }
    return 0;
}