hdu 3371 Connect the Cities

http://acm.hdu.edu.cn/showproblem.php?pid=3371

这道题要注意重边。

#include <cstdio>
#include <cstring>
#include <algorithm>
#define maxn 1000
using namespace std;
const int inf=1<<30;

int g[maxn][maxn];
int dis[maxn],a[maxn];
bool vis[maxn];
long long  sum=0;
int n,m,x,p,q,c,t;

void inti()
{
    for(int i=0; i<=n; i++)
    {
        for(int j=0; j<=n; j++)
        {
            if(i==j) g[i][j]=0;
            else
                g[i][j]=g[j][i]=inf;
        }
    }
}

void prim()
{
    memset(vis,false,sizeof(vis));
    for(int i=1; i<=n; i++) dis[i]=g[1][i];
    vis[1]=true;
    dis[1]=0;
    bool flag=false;
    for(int i=1; i<n; i++)
    {
        int mm=inf,x;
        for(int y=1; y<=n; y++) if(!vis[y]&&dis[y]<mm) mm=dis[x=y];
        vis[x]=true;
        if(mm==inf) {flag=true;break;}
        sum+=mm;
        for(int y=1; y<=n; y++) if(!vis[y]&&dis[y]>g[x][y]) dis[y]=g[x][y];
    }
    if(flag) printf("-1
");
    else
        printf("%lld
",sum);
}
int main()
{
    int k;
    scanf("%d",&k);
    while(k--)
    {
        scanf("%d%d%d",&n,&m,&x);
        inti();
        for(int i=0; i<m; i++)
        {
            scanf("%d%d%d",&p,&q,&c);
            if(c<g[p][q])
                g[p][q]=g[q][p]=c;
        }
        for(int i=0; i<x; i++)
        {
            scanf("%d",&t);
            for(int j=0; j<t; j++)
            {
                scanf("%d",&a[j]);
            }
            for(int j=0; j<t; j++)
            {
                for(int s=0; s<t; s++)
                {
                    g[a[j]][a[s]]=0;
                }
            }
        }
        sum=0;
        prim();
    }
    return 0;
}