codeforces 1076E Vasya and a Tree 樹上主席樹
阿新 • • 發佈:2018-11-22
題意:給你一棵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]); }