E. Vasya and a Tree

題意：給你一棵1為根節點的樹，初始每個節點權值為0，有m次操作，每次操作 v d x，表示將以v為根的子樹，深度不超過d的所有節點加上x。

思路：從深度為1的根節點開始遍歷，到了深度為dep的v節點，若是該節點有操作 d x，那麼以該節點為根的主席樹就要在dep+d的點加上一個權值x，然後怎麼計算該點的答案呢，該點的答案就是該點的主席樹區間 [ dep, n ]和。

#include<cstdio>
#include<algorithm>
#include<vector>
using namespace std;
typedef long long ll;
const int maxn=3e5+10;
ll sum[maxn*20],ans[maxn];
int ls[maxn*20],rs[maxn*20],cnt=0,n;
struct node
{
	int d,x;
	node(int a,int b)
	{
		d=a,x=b;
	}
};
vector<node>q[maxn];
vector<int>G[maxn];
void up(int pre,int &o,int l,int r,int k,int v)
{
	o=++cnt;
	ls[o]=ls[pre],rs[o]=rs[pre];
	sum[o]=sum[pre]+v;
	if(l==r)return;
	int m=(l+r)/2;
	if(k<=m)up(ls[pre],ls[o],l,m,k,v);
	else up(rs[o],rs[o],m+1,r,k,v);
}
ll qu(int o,int l,int r,int k)
{
	if(l>=k||!sum[o])return sum[o];
	int m=(l+r)/2;
	ll res=qu(rs[o],m+1,r,k);
	if(k<=m)res+=qu(ls[o],l,m,k);
	return res; 
}
void dfs(int u,int pre,int fa,int dep)
{
	int rt=0;
	for(int i=0;i<q[u].size();i++)
	{
		int k=min(n,q[u][i].d+dep);
		if(!rt)up(pre,rt,1,n,k,q[u][i].x);
		else up(rt,rt,1,n,k,q[u][i].x);
	}
	if(!rt)rt=pre;
	ans[u]=qu(rt,1,n,dep);
	for(int i=0;i<G[u].size();i++)
	if(G[u][i]!=fa)
	dfs(G[u][i],rt,u,dep+1);
}
int main()
{
	int u,v,x,op;
	scanf("%d",&n);
	for(int i=1;i<n;i++)
	{
		scanf("%d%d",&u,&v);
		G[u].push_back(v);
		G[v].push_back(u);
	}
	scanf("%d",&op);
	while(op--)
	{
		scanf("%d%d%d",&u,&v,&x);
		v=min(v,n);
		q[u].push_back(node(v,x));
	}
	dfs(1,0,0,1); 
	for(int i=1;i<n;i++)printf("%I64d ",ans[i]);
	printf("%I64d\n",ans[n]);
}