Gym - 101174F[(DSU)+树状数组]

版权声明：欢迎随便转载。 https://blog.csdn.net/a1214034447/article/details/89848256

题目链接:https://vjudge.net/problem/Gym-101174F

解题思路:

其实这题不同启发式合并也可以做，对rank排个序，然后在做个dfs序，把rank值小的先插入树状数组中更新，然后对每个节点查询它的dfs序的区间和就好了。

对于DSU来说就更加无脑了，本来就是"暴力"，所以我们直接DSU再加一个树状数组维护前缀和。

最坏时间复杂度就是O(n*log(n)*log(n))

#include <bits/stdc++.h>
#define fi first 
#define se second
using namespace std;
typedef long long ll;
const int N = 1e3;
const int mx = 1e5 + 10;
int n,ra[mx],ta[mx],b[mx];
int siz[mx],son[mx],m;
ll ans[mx],sum[mx];
vector <int> g[mx];
bool vis[mx];
void dfs(int u){
	siz[u] = 1,son[u] = 0;
	for(int v:g[u]){
		dfs(v);
		siz[u] += siz[v];
		if(siz[v]>siz[son[u]])
		son[u] = v;
	}
}
void add(int x,int v){
	while(x<m){
		sum[x] += v;
		x += x&(-x);
	}
}
ll getsum(int x){
	x--;ll ans = 0;
	while(x){
		ans += sum[x];
		x -= x&(-x); 
	}
	return ans;
}
void update(int u,int c){
	add(ra[u],c*ta[u]);
	for(int v:g[u]) update(v,c);	
}
void Dsu(int u,bool f){
	for(int v:g[u]){
		if(v!=son[u])
		Dsu(v,0);
	}
	if(son[u]) Dsu(son[u],1);
	for(int v:g[u]){
		if(v!=son[u])
		update(v,1);
	}
	ans[u] = getsum(ra[u]);
	add(ra[u],ta[u]);
	if(!f) update(u,-1);
}
int main()
{
	int t,rt,r;
	scanf("%d",&n);
	for(int i=1;i<=n;i++){
		scanf("%d%d%d",&m,&r,&t);
		ra[i] = r,ta[i] = t;
		b[i] = ra[i];
		if(m!=-1)
			g[m].push_back(i);
		else rt = i;
	}
	sort(b+1,b+1+n);
	m = unique(b+1,b+1+n) - b;
	for(int i=1;i<=n;i++)
	ra[i] = lower_bound(b+1,b+m,ra[i]) - b;
	dfs(rt);
	Dsu(rt,1);
	for(int i=1;i<=n;i++)
	printf("%lld\n",ans[i]);
	return 0;
}