（并查集）Travel -- hdu -- 5441（2015 ACM/ICPC Asia Regional Changchun Online ）

#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <iostream>
#include <algorithm>
using namespace std;
#define N 100100

struct node
{
    int u, v, w, index;
} a[N], b[N];

int r[N], f[N], Q[N];

int cmp(node n1, node n2)
{
    return n1.w < n2.w;
}

int Find(int x)
{
    if(x!=f[x])
        f[x]=Find(f[x]);
    return f[x];
}

int main()
{
    int t;
    scanf("%d", &t);
    while(t--)
    {
        int n, m, q, i;
        scanf("%d%d%d", &n, &m, &q);

        for(i=0; i<=n; i++)
        {
            f[i] = i;
            r[i] = 1;
        }

        for(i=0; i<m; i++)
            scanf("%d%d%d", &a[i].u, &a[i].v, &a[i].w);
        for(i=0; i<q; i++)
        {
            scanf("%d", &b[i].w);
            b[i].index = i;
        }

        sort(a, a+m, cmp);
        sort(b, b+q, cmp);

        int sum = 0, j = 0;
        for(i=0; i<q; i++)
        {
            while(j<m && a[j].w<=b[i].w)
            {
                int fu = Find(a[j].u);
                int fv = Find(a[j].v);

                if(fu!=fv)
                {
                    f[fu] = fv;
                    sum += r[fu]*r[fv]; ///两个集合的任意组合
                    r[fv] += r[fu];     ///r[i]代表i的根节点所包含的元素的个数
                }
                j++;
            }
            Q[b[i].index] = sum*2;  ///ab和ba是不一样的
        }

        for(i=0; i<q; i++)
            printf("%d\n", Q[i]);
    }
    return 0;
}