HDU 3371 Connect the Cities 最小生成数

题目地址： http://acm.hdu.edu.cn/showproblem.php?pid=3371

 

题意：

第一行代表有多少组测试数据；

接下来一行有三个数字n、m、k。n代表有多少个点，m代表下面有多少个可以修的道路，k稍后再说。

然后有m行，每行三个数字p、q、c，代表在p和q之间修一条路需要花费c。

最后还有k行，每行第一个数字n，然后这一行后面有n个数，代表这n个点之间的道路已经存在（不需要再修建）。

 

我开始的思路是将后面k行n个数之间的花费设为0，然后再用原始的Kruskal方法做，但是最后发现这样做超时了。

 

另一种思路：

将后面k行n个数之间并起来，每个构成一个连通图，然后再做。

最后判断所有节点是否都在一个集合里，如果在，说明可以修建这样的一条道路，否者不能。

 

不知道为什么这道题用G++提交超时，用C++提交就过了。

 

代码如下：

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cmath>
#include <cstring>
#include <string>
#include <algorithm>
#include <vector>
#include <set>
#include <map>
#include <queue>
#include <stack>
using namespace std;

/*
freopen("input.txt","r",stdin);
freopen("output.txt","w",stdout);
*/

int n,m,k;//n个点,总共x条边
int parent[505];//父亲节点

struct se
{
    int x,y,w;
}edge[25055];

bool emp(se a,se b)
{
    return a.w<b.w;
}

void UFset()
{
    for(int i=0;i<=n;i++)
        parent[i]=i;
}

int find(int x)
{
    int r=x;
    while(x!=parent[x])
        x=parent[x];
    while(r!=x)//剪枝优化
    {
        int j=parent[r];
        parent[r]=x;
        r=j;
    }
    return x;
}

int xiaohao()
{
    int t=find(1);
    for(int i=2;i<=n;i++)
    {
        if(t!=find(i))
            return 1;
    }
    return 0;
}

void Kruskal()
{
    int i,sum=0,count;//sum计算最小权值
    for(i=1;i<=m;i++)
    {
        int fx=find(edge[i].x);//x的根节点
        int fy=find(edge[i].y);//y的根节点
        if(fx!=fy)//两个根节点不相同代表不在一个集合里面,就合并
        {
            parent[fx]=fy;
            sum+=edge[i].w;
        }
    }
    count=xiaohao();//判断所有节点是否都在一个集合里
    if(count)
        printf("-1\n");
    else
        printf("%d\n",sum);
}

int main()
{
    int i,t;
    cin>>t;
    while(t--)
    {
        scanf("%d%d%d",&n,&m,&k);
        for(i=1;i<=m;i++)
        {
            int a,b,c;
            scanf("%d%d%d",&a,&b,&c);
            if(a>b)//保证a<b
                swap(a,b);
            edge[i].x=a;
            edge[i].y=b;
            edge[i].w=c;
        }
        UFset();//初始化父亲节点
        while(k--)
        {
            int N,xx,xy;
            cin>>N;
            cin>>xx;//第一个作为父亲节点
            for(i=1;i<N;i++)
            {
                scanf("%d",&xy);
                //parent[xy]=xx;这样写就超时了
                parent[find(xy)]=find(xx);
            }
        }
        sort(edge+1,edge+1+m,emp);//边从小到大排序
        Kruskal();
    }
    return 520;
}