codeforces 1213 G Path Queries（并查集+离线）

题目链接

小号终于上蓝了，嘿嘿。这个题比赛的时候一直在想假算法换根dp，赛后看题解没想到求联通块就完事了qwq。

题意：给你一颗树，每条边都有一个权值。q次询问，每次询问一个qi，问这颗树有多少条路径，路径上边权最大值小于qi。

题解：因为是离线查询，所以我们可以考虑离线做法。先将qi从小到大排序，然后依次更新答案。每次只将边权小于qi的边添加进图里，为啥要这样做呢，这样就能保证图里任意路径的边权最大值都小于qi。于是我们会得到一个个联通块，对于一个联通块能贡献的答案是 sz*（sz-1）/2；联通块的更新就可以用到并查集，再用sz保存每个联通块点的个数。

 

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N = 2e5+100;
int n,m;
struct node{
	int u,v,w;
}e[N];
struct Q{
	int val,id;
}q[N];
bool cmp1(node a,node b){
	return a.w<b.w;
}
bool cmp2(Q a,Q b){
	return a.val<b.val;
}
int pre[N],sz[N];
ll ans[N];
void find(int& x){
	while(x!=pre[x])
		x=pre[x]=pre[pre[x]];
}
void gkd(){
	for(int i=1;i<=n;i++) pre[i]=i,sz[i]=1;
	ll tmp=0;
	int id=1;
	for(int i=1;i<=m;i++){
		while(e[id].w<=q[i].val&&id<n){
			int x = e[id].u;
			int y = e[id].v;
			find(x);find(y);
			if(x!=y){
				tmp-=1ll*(sz[x]-1)*sz[x]/2;
				tmp-=1ll*(sz[y]-1)*sz[y]/2;
				pre[y]=x;sz[x]+=sz[y];
				tmp+=1ll*(sz[x]-1)*sz[x]/2;
			}
			id++;
		}
		ans[q[i].id]=tmp;
	}
}
int main()
{
	scanf("%d%d",&n,&m);
	for(int i=1;i<n;++i){
		int u,v,w;
		scanf("%d%d%d",&u,&v,&w);
		e[i]=node{u,v,w};
	}
	for(int i=1;i<=m;i++){
		scanf("%d",&q[i].val);
		q[i].id=i;
	}
	sort(e+1,e+n,cmp1);	
	sort(q+1,q+1+m,cmp2);
	gkd();
	for(int i=1;i<=m;++i){
		if(i!=m) printf("%lld ",ans[i]);
		else printf("%lld\n",ans[i]);
	}
		
 }