Vasya and a Tree CodeForces
阿新 • • 發佈:2018-12-21
m次修改 v k x代表將v節點之下k層的節點全都加x 最後統一查詢 即輸出每個節點的值
因為修改過程無查詢 考慮離線做法 因為是按層更新 一開始考慮維護bfs序 但k不是一個小常數 並且也無法對層數差分 只能換思路
很多資料結構問題中 區間更新點查詢和區間查詢點更新是可以相互轉換的 在這裡我就可以對每一個點 查詢會對它造成影響的所有更新的權值和 對於節點v來說 有影響的只是v到1這條鏈上的更新 並且層數還要大於等於v的層數 這樣就可以先樹剖出個dfs序 然後靜態主席樹搞個字首和 在1到v這條鏈上差分即可
#include <bits/stdc++.h> using namespace std; #define pb push_back typedef long long ll; const int maxn=3e5+10; struct node0 { ll val; int k; }; struct node1 { int v; int next; }; struct node2 { int l; int r; ll val; }; vector <node0> pre[maxn]; node1 edge[2*maxn]; node2 tree[40*maxn]; ll ans[maxn]; int first[maxn],fa[maxn],deep[maxn],sum[maxn],son[maxn],top[maxn],mp1[maxn],mp2[maxn],root[maxn]; int n,q,num; void addedge(int u,int v) { edge[num].v=v; edge[num].next=first[u]; first[u]=num++; return; } void dfsI(int cur) { int i,v; sum[cur]=1,son[cur]=-1; for(i=first[cur];i!=-1;i=edge[i].next){ v=edge[i].v; if(v!=fa[cur]){ fa[v]=cur,deep[v]=deep[cur]+1; dfsI(v); sum[cur]+=sum[v]; if(son[cur]==-1||sum[son[cur]]<sum[v]){ son[cur]=v; } } } } void dfsII(int cur,int tp) { int i,v; num++; top[cur]=tp,mp1[cur]=num,mp2[num]=cur; if(son[cur]==-1) return; dfsII(son[cur],tp); for(i=first[cur];i!=-1;i=edge[i].next){ v=edge[i].v; if(v!=fa[cur]&&v!=son[cur]){ dfsII(v,v); } } } int build(int l,int r) { int cur,m; cur=num++; tree[cur].l=0,tree[cur].r=0,tree[cur].val=0; if(l==r) return cur; m=(l+r)/2; tree[cur].l=build(l,m); tree[cur].r=build(m+1,r); return cur; } int update(int rot,int tar,ll val,int l,int r) { int cur,m; cur=num++; tree[cur]=tree[rot]; tree[cur].val+=val; if(l==r) return cur; m=(l+r)/2; if(tar<=m) tree[cur].l=update(tree[rot].l,tar,val,l,m); else tree[cur].r=update(tree[rot].r,tar,val,m+1,r); return cur; } ll query(int lrot,int rrot,int pl,int pr,int l,int r) { ll res; int m; if(pl<=l&&r<=pr) return tree[rrot].val-tree[lrot].val; res=0,m=(l+r)/2; if(pl<=m) res+=query(tree[lrot].l,tree[rrot].l,pl,pr,l,m); if(pr>m) res+=query(tree[lrot].r,tree[rrot].r,pl,pr,m+1,r); return res; } ll solve(int u,int v,int tar) { ll res; //printf("***%d***\n",tar); res=0; while(top[u]!=top[v]){ if(deep[top[u]]<deep[top[v]]) swap(u,v); //printf("*%d %d*\n",top[u],u); res+=query(root[mp1[top[u]]-1],root[mp1[u]],deep[tar],n,1,n); u=fa[top[u]]; } if(deep[u]<deep[v]) swap(u,v); //printf("*%d %d*\n",v,u); res+=query(root[mp1[v]-1],root[mp1[u]],deep[tar],n,1,n); return res; } void show(int rot,int l,int r) { int m; printf("*%lld %d %d*\n",tree[rot].val,l,r); if(l==r) return; m=(l+r)/2; show(tree[rot].l,l,m); show(tree[rot].r,m+1,r); } int main() { node0 tmp; int i,j,u,v,flag; scanf("%d",&n); memset(first,-1,sizeof(first)); num=0; for(i=1;i<=n-1;i++){ scanf("%d%d",&u,&v); addedge(u,v); addedge(v,u); } fa[1]=0,deep[1]=1; dfsI(1); num=0; dfsII(1,1); scanf("%d",&q); while(q--){ scanf("%d%d%lld",&v,&tmp.k,&tmp.val); pre[mp1[v]].pb(tmp); } num=0; root[0]=build(1,n); for(i=1;i<=n;i++){ flag=0; //printf("***%d %d***\n",i,deep[i]); for(j=0;j<pre[i].size();j++){ //printf("*%d %lld*\n",pre[i][j].k,pre[i][j].val); if(flag==0){ root[i]=update(root[i-1],min(n,deep[mp2[i]]+pre[i][j].k),pre[i][j].val,1,n); flag=1; } else{ root[i]=update(root[i],min(n,deep[mp2[i]]+pre[i][j].k),pre[i][j].val,1,n); } } if(flag==0) root[i]=root[i-1];// } /* for(i=1;i<=n;i++){ printf("***%d***\n",i); show(root[i],1,n); } */ for(i=1;i<=n;i++){ ans[i]=solve(1,i,i); } for(i=1;i<=n;i++) printf("%lld ",ans[i]); printf("\n"); return 0; }