Codeforces 1076E E. Vasya and a Tree

这是一个静态问题
按DFS走，进入一个子树时把子树根上挂的修改激活，离开子树时撤销
这样所有的修改和查询都是对于深度的

QvQ：递归三参数栈爆炸了

#include<bits/stdc++.h>
using namespace std;
#define M 524288
#define N 300005
typedef long long ll;

int L,R,aim,n,m,tot,s[N],dep;
ll t[M<<1],ans,op[N];
vector<pair<int,int> > q[N];

struct edge{int v,n;}e[N<<1];
#define Push(u,v) push(u,v),push(v,u)
inline void push(int u,int v){e[++tot]=(edge){v,s[u]};s[u]=tot;}

void change(int k,int l,int r)
{
	if (L<=l && r<=R){t[k]+=aim;return;}
	int mid=l+r>>1;
	if (L<=mid) change(k<<1,l,mid);
	if (mid<R) change(k<<1|1,mid+1,r);
}

void find(int k,int l,int r)
{
	ans+=t[k];
	if (l==r) return;
	int mid=l+r>>1;
	if (aim<=mid) find(k<<1,l,mid);
	else find(k<<1|1,mid+1,r);
}

void dfs(int k,int f)
{
	dep++; 
	for (auto s:q[k])
	{
		L=dep;R=dep+s.first;
		aim=s.second;
		change(1,1,M);
	}
	aim=dep;ans=0;
	find(1,1,M);
	op[k]=ans;
	for (int i=s[k];i;i=e[i].n) if (e[i].v!=f) dfs(e[i].v,k);
	for (auto s:q[k])
	{
		L=dep;R=dep+s.first;
		aim=-s.second;
		change(1,1,M);
	}
	dep--;
}

int main()
{
	scanf("%d",&n);
	for (int i=1,u,v;i<n;i++) scanf("%d%d",&u,&v),Push(u,v);
	scanf("%d",&m);
	for (int v,d,x;m--;)
	{
		scanf("%d%d%d",&v,&d,&x);
		q[v].push_back(make_pair(d,x));
	}
	dfs(1,0);
	for (int i=1;i<=n;i++) printf("%I64d%c",op[i],i==n?'\n':' ');
}