HDU 5441 Travel(2015 ACM/ICPC Asia Regional Changchun Online)

题目分析

这道理第一感觉就是暴力，但是数据量实在太大，但是我们可以有策略的暴力，这道题我的方法就是先排序再暴力，首先我们可以对于距离进行排序，然后我们离线处理q次询问，同样的方法对于q此询问的距离进行排序（注意存标号）。然后用并查集处理即可，即可方法见代码。

#include <cstdio>
#include <vector>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
const int maxn = 1e5+100;

struct Edge{
    int from, to, dist;
    bool operator < (const Edge temp)const{
        return dist < temp.dist;
    }
}e[maxn];
int fa[20005], sum[20005], ans[5005];

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

struct Node{
    int id, d;
    bool operator < (const Node temp)const{
        return d < temp.d;
    }
}node[5005];

void init(){
    for(int i = 0; i < 20005; i++) fa[i] = i, sum[i] = 1;
}

int main(){
    int T, n, m, q;
    scanf("%d", &T);
    while(T--){
        init();
        scanf("%d%d%d", &n, &m, &q);
        for(int i = 0; i < m; i++)
            scanf("%d%d%d", &e[i].from, &e[i].to, &e[i].dist);
        sort(e, e+m);
        for(int i = 0; i < q; i++){
            node[i].id = i;
            scanf("%d", &node[i].d);
        }
        sort(node, node+q);
        int j = 0;
        for(int i = 0; i < q; i++){
            for(; e[j].dist <= node[i].d && j < m; j++){
                int xx = Find(e[j].from);
                int yy = Find(e[j].to);
                if(xx != yy){
                    fa[xx] = yy;
                    sum[yy] += sum[xx];
                    sum[xx] = 0;
                }
            }
            int cnt = 0;
            for(int k = 1; k <= n; k++){
                if(fa[k] == k)
                    cnt += (sum[k]*(sum[k]-1));
            }
            ans[node[i].id] = cnt;
        }
        for(int i = 0; i < q; i++)
            printf("%d\n", ans[i]);
    }
    return 0;
}