HDU 3371 Connect the Cities 【kruskal】

来源：互联网发布：歼31和f35对比数据编辑：程序博客网时间：2024/05/21 15:46

点击打开链接

题意：给你n个城市让你花最小代价连起来，但是给你了几个已经连好的城市。

我上来怼了一发prim 超时，我就想，kruskal 应该可以，最后没时间敲，，，gg

直接kruskal 模板就能过。

#include <stdio.h>#include <string.h>#include <iostream>#include <algorithm>#include <queue>using namespace std;const int inf=0xffffff;const int maxn=505;struct node{    int u,v,w;    bool operator <(const node &a)const{        return w<a.w;    }}edge[maxn*maxn];int n,m,w,T,cnt,t;int fa[maxn];int dis[maxn];void addedge(int x,int y,int val){    edge[++cnt].u=x;    edge[cnt].v=y;    edge[cnt].w=val;}int Find(int x){    if(fa[x]==-1) return x;    return fa[x]=Find(fa[x]);}int kruskal(int n){    int ans=0;    for(int i=1;i<=cnt;++i){        int u=edge[i].u;        int v=edge[i].v;        int w=edge[i].w;        int t1=Find(u);        int t2=Find(v);        if(t1!=t2){            ans+=w;            fa[t1]=t2;            t++;        }        if(t==n) break;    }    if(t<n) return -1;    else return ans;}int main(){    int a,b,c,q;    scanf("%d",&T);    while(T--){        cnt=0;t=1;        memset(fa,-1,sizeof(fa));        scanf("%d %d %d",&n,&m,&w);        while(m--){            scanf("%d %d %d",&a,&b,&c);            addedge(a,b,c);            addedge(b,a,c);        }        sort(edge+1,edge+1+cnt);        while(w--){            scanf("%d",&q);            for(int i=0;i<q;++i){                scanf("%d",&a);                if(!i) b=a;                else {                    int t1=Find(a),t2=Find(b);                    if(t1!=t2) fa[t1]=t2,t++;                    b=a;                }            }        }        int ans=kruskal(n);        printf("%d\n",ans);    }    return 0;}

0 0