hdu 3371 Connect the Cities 最小生成树(kruskal算法)

#include <cstdio>
#include <cstring>
#include <cmath>
#include <vector>
#include <iostream>
#include <algorithm>
using namespace std;
const int maxn=505;
const int maxm=25005;
int f[maxn];
struct node{
    int a,b,w;
}e[maxm];
int find(int x)
{
    if(x!=f[x])f[x]=find(f[x]);
    return f[x];
}
int cmp(node a,node b)
{
    return a.w<b.w;
}
int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        int i,j,k,n,m,x,y,z,num=0,ans=0,aa,bb,t;
        scanf("%d%d%d",&n,&m,&k);
        for(i=1;i<=n;i++)
            f[i]=i;
        for(i=0;i<m;i++)
        {
            scanf("%d%d%d",&x,&y,&z);
            e[i].a=x;
            e[i].b=y;
            e[i].w=z;
        }
        sort(e,e+m,cmp);
        //cout<<k<<"***"<<endl;
        for(i=0;i<k;i++)
        {
            scanf("%d",&t);
            scanf("%d",&x);
            for(j=1;j<t;j++)
            {
                scanf("%d",&y);
                aa=find(x);
                bb=find(y);
                x=y;
                if(aa!=bb)
                {
                    f[aa]=bb;
                    num++;
                }
            }
        }
        if(num==n-1){printf("0\n");continue;}
        for(i=0;i<m;i++)
        {
            aa=find(e[i].a);
            bb=find(e[i].b);
            if(aa==bb)continue;
            f[aa]=bb;
            num++;
            ans+=e[i].w;
            if(num==n-1)break;
        }
        if(num==n-1)
            printf("%d\n",ans);
        else
            printf("-1\n");
    }
    return 0;
}
/*
    最小生成树,kruskal算法
*/

你可能感兴趣的:(最小生成树)