CF 1076 E

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/fufck/article/details/84197408

#include<cstdio>
#include<vector>
using namespace std;
typedef long long ll;
#define low(i) ((i)&(-i))
const int maxn=3e5+5;
int n,m,tot;
ll val[maxn];
int dep[maxn],now[maxn],pre[maxn*2],son[maxn*2];

int read() {
    int x=0,f=1;char ch=getchar();
    for(;ch<'0'||ch>'9';ch=getchar())if(ch=='-')f=-1;
    for(;ch>='0'&&ch<='9';ch=getchar())x=x*10+ch-'0';
    return x*f;
}

struct query
{
	int v,d;
	query(){}
	query(int _v,int _d){
		v=_v;d=_d;
	}
} ;

vector<query> s[maxn];
struct TreeArray{
	ll c[maxn];
	void add(int pos,int v)
	{
		for(int i=pos;i<=n;i+=low(i)) c[i]+=v;
	}
	ll query(int pos)
	{
		ll res=0;
		for(int i=pos;i;i-=low(i)) res+=c[i];
		return res;
	}
}T;

void add(int a,int b)
{
	pre[++tot]=now[a];
	now[a]=tot;son[tot]=b;
}

void dfs(int fa,int u)
{
	dep[u]=dep[fa]+1;
	for(int p=now[u],v=son[p];p;p=pre[p],v=son[p])
	     if(v!=fa) dfs(u,v);
}
void make_ans(int fa,int u)
{
	vector<query>::iterator it;
	for(it=s[u].begin();it!=s[u].end();it++)
	{
		int l=dep[u],r=(*it).d+dep[u];r=min(r,n);
		T.add(l,(*it).v);T.add(r+1,-(*it).v);
	}
	val[u]=T.query(dep[u]);
	for(int p=now[u],v=son[p];p;p=pre[p],v=son[p])
		if(v!=fa) make_ans(u,v);
	for(it=s[u].begin();it!=s[u].end();it++)
	{
		int l=dep[u],r=(*it).d+dep[u];r=min(r,n);
		T.add(l,-(*it).v);T.add(r+1,(*it).v);
	}
}
int main()
{
	n=read();
	for(int i=1;i<n;i++)
	{
	   int a=read(),b=read();
	   add(a,b);add(b,a); 
	}
	dfs(0,1);m=read();
	for(int i=1;i<=m;i++)
	{
		int u=read(),d=read(),x=read();
		s[u].push_back(query(x,d));
	}
	make_ans(0,1);
	for(int i=1;i<=n;i++) printf("%I64d ",val[i]);
	return 0;
}