Codeforces 1213G

一个有向图，每条边都有一个权值，有m个询问，每个询问找出有多少对点使得这些点对之间的最大边权不大于q
直接建图搞不好做，也许是我不会吧
考虑要查询的值为x,当前有一条边，其权值为x-1,连接的两个连通分量大小为sz1和sz2，那么很显然这条边的贡献就是sz1*sz2
而权值小的边一直都对更大的查询有贡献。
从而想到，可以讲边和查询排序，然后用并查集维护连通分量的大小。

#include<bits/stdc++.h>
using namespace std;
#define f first
#define s second
typedef long long LL;
typedef pair<int,int> PII;
typedef pair<LL,LL> PLL;
const int N = 2e5 + 10;
int fa[N],sz[N],n,m,a,b,c;
struct query{
	int val,id;
	LL ans;
}q[N];
struct edge{
	int u,v,w;
}E[N];
int cmp1(const query & x, const query & y){
	return x.val < y.val;
}
int cmp2(const query & x, const query & y){
	return x.id < y.id;
}
int cmp3(const edge & x, const edge & y){
	return x.w < y.w;
}
int find(int x){
	return x == fa[x] ? x : fa[x] = find(fa[x]);
}
void Union(int u,int v){
	u = find(u);
	v = find(v);
	fa[v] = u;
	sz[u] += sz[v];
	sz[v] = sz[u];
}
int main(){
	ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
	//freopen("data.in","r",stdin);
	//freopen("data.out","w",stdout);
	cin>>n>>m;
	for(int i = 1 ; i <= n ; i++) fa[i] = i, sz[i] = 1;
	for(int i = 0 ; i < n - 1; i ++) {
		cin>>E[i].u>>E[i].v>>E[i].w;
	}
	sort(E,E+n-1,cmp3);
	for(int i = 0 ; i < m; i ++) {
		cin>>q[i].val;
		q[i].id = i;
	}
	sort(q,q+m,cmp1);
	int i = 0, j = 0;
	LL res = 0 ;
	q[0].ans = 0 ;
	while(j < m){
		if(j) q[j].ans = q[j-1].ans;
		while(E[i].w <= q[j].val && i < n - 1 ){
			int u = E[i].u , v = E[i].v;
			q[j].ans += (LL)sz[find(u)] * sz[find(v)];
			Union(u,v);
			i++;
		}
		j++;
	}
	sort(q,q+m,cmp2);
	for(int i = 0 ; i < m ;i ++) cout<<q[i].ans<<' ';cout<<endl;
	return 0;
}