CF1213G Path Queries

题目传送门

解题方法

本题可以把查询离线，然后我们把边权从小到大排序，每次加入完这个权值的边更新这个查询的答案，同时更新后面的答案。

然后如何更新答案呢，我们只需要求出来需要练边的两侧点分别联通了多少个点，然后加上两边联动点的量的积。这个东西该如何储存呢？我们只是需要一个假一点的权值并差集，因为只查询这两个点的根的节点有多少个节点，所以只需要把答案存在根上就可以了。

代码：

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
struct node
{
	long long x,y,v;
}a[400010];
long long ans[400010];
bool comp(const node &xx,const node &yy)
{
	return xx.v<yy.v; 
} 
long long fa[400010],he[400010];
void pre(long long x)
{
	for(long long i=1;i<=x;i++)
	  fa[i]=i,he[i]=1;
}
long long find(long long x)
{
	if(x!=fa[x])return fa[x]=find(fa[x]);
	else return x;
}
void merge(long long x,long long y)
{
	x=find(x);
	y=find(y);
	he[x]+=he[y]; 
	fa[y]=x;
}
int main()
{
	long long n,m;
	scanf("%lld%lld",&n,&m);
	pre(n); 
	for(long long i=1;i<n;i++)
	{
		scanf("%lld%lld%lld",&a[i].x,&a[i].y,&a[i].v);
	}
	sort(a+1,a+n,comp);
	for(long long i=1;i<n;i++)
	{
		ans[a[i].v]+=he[find(a[i].x)]*he[find(a[i].y)];
		merge(a[i].x,a[i].y);
	}
	for(long long i=1;i<=2e5;i++)
	  ans[i]+=ans[i-1];
	for(long long i=1;i<=m;i++)
	{
		long long x;
		cin>>x;
		cout<<ans[x]<<"\n";
	}
	return 0;
}