HDU-3371 Connect the Cities(最小生成树)

最小生成树

卡常数的水题,注意下细节就可以

#include<cstdio>
#include<algorithm>
#include<string.h>
using namespace std;
const int maxn = 500 + 5;
int p[maxn],c[maxn];
struct Road{
    int x,y,cost;
    bool operator<(const Road& _A)const{
        return cost<_A.cost;
    }
}r[maxn*maxn];
int find(int x){
    return p[x]==x?x:(p[x]=find(p[x]));
}
int main(){
    int i,T,n,m,k,t,p1,p2,ans,cnt;
   // freopen("in.txt","r",stdin);
    scanf("%d",&T);
    while(T--){
        scanf("%d%d%d",&n,&m,&k);
        for(i=1;i<=n;i++) p[i]=i;
        for(i=0;i<m;i++) scanf("%d%d%d",&r[i].x,&r[i].y,&r[i].cost);
        sort(r,r+m);
        while(k--){
            scanf("%d%d",&t,&c[0]);
            p1=find(c[0]);
            for(i=1;i<t;i++) {
                scanf("%d",&c[i]);
                p2=find(c[i]);
                if(p1!=p2) p[p2]=p1;
            }
        }
        ans=0;
        for(i=0;i<m;i++){
            p1=find(r[i].x),p2=find(r[i].y);
            if(p1!=p2) {
                ans+=r[i].cost;
                p[p2]=p1;
            }
        }
        cnt=0;
        for(i=1;i<=n;i++) if(find(i)==i) cnt++;
        if(cnt>1) printf("-1\n");
        else printf("%d\n",ans);
    }
    return 0;
}