解题思路:
是对子树的操作,容易想到dfs序,但又是对深度小于等于k的子树操作,在dfs序中不是连续的,如果考虑bfs序,深度是递增的,但是连续的段又不满足是子树,也不行。
本题的特点是查询只有一次,故可以离线,对于任意一个点,只有他的所有父会产生影响,还是考虑dfs ,这样每遍历一个点他的所有父一定已经访问过了,用数据结构维护一个深度段(树状数组,线段树都可),每访问到一个点,把他的操作全做一边,(区间加操作,经典差分)。此时,这个点的答案直接在数据结构里查询就可以。(影响他的点已经全操作过了)
仅仅这样还有一个问题,数据结构维护的是深度,每次子树访问结束时,要回退所有操作。(举个最简单的例子,根结点的左右子树有相同的深度,如果不回退就会产生影响)
Code:
#include <iostream>
#include <map>
#include <set>
#include <queue>
#include <stack>
#include <algorithm>
#include <vector>
#include <string>
#include <iomanip>
#include <cmath>
#include <ctime>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <climits>
//#include <unordered_map>
#define guo312 std::ios::sync_with_stdio(false), cin.tie(0), cout.tie(0)
#define ll long long
#define Inf LONG_LONG_MAX
#define inf INT_MAX
#define endl "\n"
#define PI 3.1415926535898
using namespace std;
const int N=3e5+10;
int n,m;
vector<int> ve1[N];
vector<pair<int,int>> ve2[N];
ll ans[N],c[N];
int lowbit(int x){
return x&(-x);
}
void add(int x,int num){
for( ;x<=n;c[x]+=num,x+=lowbit(x));
}
ll query(int x){
ll re=0;
for( ;x>0;re+=c[x],x-=lowbit(x));
return re;
}
void dfs(int u,int fa,int dep){
for(auto it:ve2[u]){
add(dep,it.second);
add(dep+it.first+1,-it.second);
}
ans[u]=query(dep);
for(auto it:ve1[u]){
if(it!=fa){
dfs(it,u,dep+1);
}
}
for(auto it:ve2[u]){
add(dep,-it.second);
add(dep+it.first+1,it.second);
}
}
int main(){
guo312;
cin>>n;
for(int i=1;i<n;i++){
int s1,s2; cin>>s1>>s2;
ve1[s1].push_back(s2);
ve1[s2].push_back(s1);
}
cin>>m;
for(int i=1;i<=m;i++){
int s1,s2,s3; cin>>s1>>s2>>s3;
ve2[s1].push_back({s2,s3});
}
dfs(1,0,1);
for(int i=1;i<=n;i++){
cout<<ans[i]<<" ";
}
return 0;
}