hdu-3371 Connect the Cities---kruskal

题目链接：

http://acm.hdu.edu.cn/showproblem.php?pid=3371

题目大意：

给n个城市，m条路，k组已知路，求最小费用联通所有城市；

解题思路：

kruskal求MST，这道题目有毒，很容易超时，改了一下并查集才过，而且同一个代码有时过又是超时

 

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn = 500 + 10;
const int INF = 0x3f3f3f3f;
int p[maxn];
int tot;
void init()
{
    for(int i = 0; i < maxn; i++)p[i] = i;
}
int f(int x)
{
    return x == p[x] ? x : p[x] = f(p[x]);
}
void Union(int x, int y)
{
    x = f(x);
    y = f(y);
    if(x != y)
    {
        p[x] = y;//这里改成p[y] = x;有时过，有时会超时
        tot--;//不联通的数目减一
    }
}
struct edge
{
    int u, v, w;
    bool operator < (const edge& a)
    {
        return w < a.w;
    }
}a[maxn * maxn];
int main()
{
    int T, n, m, k, t, x, y;
    scanf("%d", &T);
    while(T--)
    {
        init();
        scanf("%d%d%d", &n, &m, &k);
        tot = n - 1;//最开始，不连通块数目为n-1
        for(int i = 1; i <= m; i++)
        {
            scanf("%d%d%d", &a[i].u, &a[i].v, &a[i].w);
        }
        sort(a + 1, a + 1 + m);
        while(k--)
        {
            scanf("%d%d", &t, &x);
            t--;
            while(t--)
            {
                scanf("%d", &y);
                Union(x, y);
            }
        }
        int ans = 0;/*
        for(int i = 1; i <= n; i++)cout<<p[i]<<" ";
        cout<<endl;*/
        for(int i = 1; i <= m; i++)
        {
            int x = a[i].u;
            int y = a[i].v;
            x = f(x), y = f(y);
            if(x != y)
            {
                p[x] = y;
                tot--;//不连通数目减一
                ans += a[i].w;
            }
        }
        if(tot == 0)//全部连通
            printf("%d\n", ans);
        else
            printf("-1\n");
    }
    return 0;
}