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;
}