题意:给你一个带权的无向图,然后q(q≤5000)次询问,问有多少对城市(城市对(u,v)与(v,u)算不同的城市对,而且u≠v)之间的边的长度不超过d(如果城市u到城市v途经城市w,
那么需要城市u到城市w的长度e1≤d,同时城市w到城市v的长度e2≤d)。
析:一开始的时候,题意都读错了,怎么看都不对,原来是只要最大的d小于等于x就可以,过了好几天才知道是这样。。。。。
这个题是并查集的应用,先从d小的开始遍历,然后去判断有几个连通块,在连通块中计数,用一个数组即可,运用排列组合的知识可以知道一个连通块中, 一共有
n * (n-1)对,然后不断更新连通块中结点的数量即可,先排序再输出。其实并不难。要注意加结点的时候,谁是谁是父结点,这个是很重要的。
代码如下:
#pragma comment(linker, "/STACK:1024000000,1024000000")
#include <cstdio>
#include <string>
#include <cstdlib>
#include <cmath>
#include <iostream>
#include <cstring>
#include <set>
#include <queue>
#include <algorithm>
#include <vector>
#include <map>
#include <cctype>
using namespace std ; typedef long long LL;
typedef pair<int, int> P;
const int INF = 0x3f3f3f3f;
const double inf = 0x3f3f3f3f3f3f3f;
const double eps = 1e-8;
const int maxn = 2e4 + 5;
const int mod = 1e9 + 7;
const int dr[] = {0, 0, -1, 1};
const int dc[] = {-1, 1, 0, 0};
int n, m;
inline bool is_in(int r, int c){
return r >= 0 && r < n && c >= 0 && c < m;
}
struct node{
int u, v, w;
bool operator < (const node &p) const{
return w < p.w;
}
};
node a[maxn*5];
struct node1{
int id, w;
bool operator < (const node1 &p) const{
return w < p.w;
}
};
node1 x[5005];
int p[maxn], w[maxn], ans[maxn]; int Find(int x){ return x == p[x] ? x : p[x] = Find(p[x]); } int main(){
int T; cin >> T;
while(T--){
int q;
scanf("%d %d %d", &n, &m, &q);
for(int i = 1; i <= n; ++i) p[i] = i, w[i] = 1;
for(int i = 0; i < m; ++i) scanf("%d %d %d", &a[i].u, &a[i].v, &a[i].w);
for(int i = 0; i < q; ++i) scanf("%d", &x[i].w), x[i].id = i;
sort(a, a+m);
sort(x, x+q); int j = 0, num = 0;
for(int i = 0; i < q; ++i){
while(j < m && a[j].w <= x[i].w){
int xx = Find(a[j].u);
int yy = Find(a[j].v);
if(xx != yy){
num -= w[xx] * (w[xx]-1) + w[yy] * (w[yy]-1);
w[xx] += w[yy];
w[yy] = 0;
num += w[xx] * (w[xx]-1);
p[yy] = xx;//注意父结点的选取
}
++j;
}
ans[x[i].id] = num;
} for(int i = 0; i < q; ++i)
printf("%d\n", ans[i]);
}
return 0;
}