hdu5441Travel【并查集】

大意：告诉你有n个点  m个边的无向图  

然后问有多少点对  他们的路径上节点之间的距离都少于 x

分析:并查集

代码:

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;

const long long maxn = 1000005;
struct E {
    long long u, v, w;
}e[maxn];
bool cmp(E e1, E e2) {
    if(e1.w != e2.w) {
        return e1.w < e2.w;
    }
    return e1.u < e2.u;
}
struct ANS {
    long long id;
    long long num;
}ans[maxn];
bool cmp2(ANS a1, ANS a2) {
    if(a1.num != a2.num) {
        return a1.num < a2.num;
    }
    return a1.id < a2.id;
}

long long fa[maxn];
long long xx[maxn];
long long find(long long x) {
    if(x == fa[x]) {
        return x;
    }
    return fa[x] = find(fa[x]);
}
void init(long long n) {
    for(long long i = 0; i <= n; i++) {
        fa[i] = i;
        xx[i] = 1;
    }
}
void unin(long long x, long long y) {
    long long fx = find(x), fy = find(y);
    if(fx != fy) {
        fa[fx] = fy;
        xx[fy] += xx[fx];
        xx[fx] = 0;
    }
}

long long val[maxn];

int main() {
    long long t, n, m, q;
    scanf("%I64d",&t);
    while(t--) {
        scanf("%I64d%I64d%I64d",&n, &m, &q);
        memset(xx, 0, sizeof(xx));
        init(n);
        for(long long i = 0; i < m; i++) {
            scanf("%I64d%I64d%I64d",&e[i].u, &e[i].v, &e[i].w);
        }
        sort(e, e + m, cmp);
        for(long long i = 0; i < q; i++) {
            scanf("%I64d",&ans[i].num);
            ans[i].id = i;
        }
        sort(ans, ans + q, cmp2);
        long long i = 0, j = 0;
        long long _ans = 0;
        while(i < m && j < q) {
            while(i < m && j < q && e[i].w <= ans[j].num) {
                long long x = find(e[i].u), y = find(e[i].v);
                if(x != y) {
                    _ans += xx[x] * xx[y];
                    unin(x, y);
                }
                i++;
            }
            val[ans[j++].id] = _ans;
        }
        while(j < q) {
            val[ans[j++].id] = _ans;
        }
        for(long long i = 0; i < q; i++) {
            printf("%I64d
", val[i] * 2);
        }
    }
}