hdu3371 题目是不是变态,我就不评论了,不过我的确学了不少东西 如果让我说:我只能说,实力决定一切

思路:刚开始想把已经修好的路,放到同一集合中去,不过发现这样做在查找根节点时,会出错,我就把已修好的路的花费设为0,放到优先队列中,不过这样挺浪费时间滴。

C++提交卡过,G++, TL

 

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<queue>
using namespace std;
struct node
{
    int st,ed,w;
    bool friend operator <(const node &a,const node &b)
    {
        return a.w>b.w;
    }
};
int far[511];
int rank[511];
int n,m,k;
int s;
priority_queue<node>Q;
void init()
{
    for(int i=0; i<=n; i++)
    {
        far[i]=i;
        rank[i]=1;
    }
}
int find_set(int x)
{
    if(far[x]==x)
        return x;
    far[x]=find_set(far[x]);
    return far[x];
}
int kruskal(int nn)
{
    int sum=0;
    node  tem;
    while(!Q.empty())
    {
        tem=Q.top();
        Q.pop();
        int x=find_set(tem.st);
        int y=find_set(tem.ed);
        if(x!=y)
        {
            if(rank[x]>rank[y])
            {
                far[y]=x;
                rank[x]+=rank[y];
            }
            else
            {
                far[x]=y;
                rank[y]+=rank[x];
            }
            sum+=tem.w;
            s++;
        }
    }
    return sum;
}
int main()
{
    int T;
    int p,q,c;
    node tem;
    scanf("%d\n",&T);
    while(T--)
    {
        while(!Q.empty())
            Q.pop();
        s=0;
        scanf("%d%d%d",&n,&m,&k);
        init();
        for(int i=0; i<m; i++)
        {
            scanf("%d%d%d",&p,&q,&c);
            tem.st=p;
            tem.ed=q;
            tem.w=c;
            Q.push(tem);
        }
        int gh;
        for(int i=0; i<k; i++)
        {
            scanf("%d",&gh);
            scanf("%d",&p);
            for(int j=1; j<gh; j++)
            {
                scanf("%d",&q);
                tem.st=p;
                tem.ed=q;
                tem.w=0;
                Q.push(tem);
            }
        }
        int ans=kruskal(n);
        if(s!=n-1)
            printf("-1\n");
        else
            printf("%d\n",ans);
    }
    return 0;
}

你可能感兴趣的:(hdu3371 题目是不是变态,我就不评论了,不过我的确学了不少东西 如果让我说:我只能说,实力决定一切)