E. Vasya and a Tree
题意
做法
代码
#include<stdio.h>
#include<iostream>
#include<algorithm>
#include<vector>
using namespace std;
typedef long long ll;
const int maxn = 3e5+10;
int n,m;
ll tree[maxn],ans[maxn];
vector<int> G[maxn],D[maxn],X[maxn];
void add(int x,int val)
{
while(x<=n)
{
tree[x]+=val;
x=x+(x&-x);
}
}
ll sum(int x)
{
ll ans=0;
while(x)
{
ans+=tree[x];
x=x-(x&-x);
}
return ans;
}
void dfs(int x,int fa,int dep)
{
for(int i=0;i<D[x].size();i++)
{
add(min(D[x][i]+dep,n),X[x][i]);//进子树之前更新
}
ans[x]=sum(n)-sum(dep-1);//树状数组变区间查询为两个前缀和相减
//由于性质2,所以在这个地方就可以直接算出当前节点的最终答案
for(int i=0;i<G[x].size();i++)
{
if(G[x][i]==fa) continue;
dfs(G[x][i],x,dep+1);
}
for(int i=0;i<D[x].size();i++)
{
add(min(D[x][i]+dep,n),-X[x][i]);//出子树之后还原
}
}
int main()
{
int x,y,z;
scanf("%d",&n);
for(int i=1;i<=n-1;i++)
{
scanf("%d%d",&x,&y);
G[x].push_back(y);
G[y].push_back(x);
}
scanf("%d",&m);
for(int i=1;i<=m;i++)
{
scanf("%d%d%d",&x,&y,&z);
D[x].push_back(y);
X[x].push_back(z);
}
dfs(1,0,1);
for(int i=1;i<=n;i++)
{
printf("%lld ",ans[i]);
}
return 0;
}