POJ 1007 WZJ的数据结构（七） 树上操作

传送门：http://oj.cnuschool.org.cn/oj/home/problem.htm?problemID=983

树链剖分版：

  1 #include<iostream>
  2 #include<cstdio>
  3 #include<cmath>
  4 #include<algorithm>
  5 #include<queue>
  6 #include<cstring>
  7 #define PAU putchar(‘ ‘)
  8 #define ENT putchar(‘\n‘)
  9 using namespace std;
 10 const int maxn=100000+10,inf=-1u>>1,maxn3=3*maxn;
 11 struct Tedge{int x,y,w,next;}adj[maxn*2];int ms=0,fch[maxn];
 12 struct Edge{int from,to,dist;}e[maxn];
 13 void AddEdge(int u,int v,int w){adj[++ms]=(Tedge){u,v,w,fch[u]};fch[u]=ms;return;}
 14 int top[maxn],dep[maxn],son[maxn],siz[maxn],fa[maxn],sumv[maxn3],addv[maxn3],num[maxn],sz=0,ql,qr,cv,_sum,n,Q;
 15 void dfs(int u){
 16     siz[u]=1;dep[u]=dep[fa[u]]+1;
 17     for(int i=fch[u];i;i=adj[i].next){
 18         int v=adj[i].y;
 19         if(v!=fa[u]){
 20             fa[v]=u;
 21             dfs(v);
 22             if(siz[son[u]]<siz[v]) son[u]=v;
 23             siz[u]+=siz[v];
 24         }
 25     } return;
 26 }
 27 void build(int u,int tp){
 28     num[u]=++sz;top[u]=tp;
 29     if(son[u]) build(son[u],tp);
 30     for(int i=fch[u];i;i=adj[i].next){
 31         int v=adj[i].y;
 32         if(v!=fa[u]&&v!=son[u]) build(v,v);
 33     } return;
 34 }
 35 void update(int o,int L,int R){
 36     if(ql<=L&&R<=qr) addv[o]+=cv;
 37     else{
 38         int M=L+R>>1,lc=o<<1,rc=lc|1;
 39         if(ql<=M) update(lc,L,M);
 40         if(qr>M) update(rc,M+1,R);
 41     } return;
 42 }
 43 void change(int a,int b){
 44     _sum=0;
 45     int f1=top[a],f2=top[b];
 46     while(f1!=f2){
 47         if(dep[f1]<dep[f2]) swap(f1,f2),swap(a,b);
 48         ql=num[f1];qr=num[a];update(1,1,n);
 49         a=fa[f1];f1=top[a];
 50     }
 51     if(dep[a]>dep[b]) swap(a,b);
 52     ql=num[a];qr=num[b];update(1,1,n);return;//为毛不是孩子了
 53 }
 54 void clear_set(int o,int L,int R,int add){
 55     if(L==R) sumv[L]+=add+addv[o];
 56     else{
 57         int M=L+R>>1,lc=o<<1,rc=lc|1;
 58         clear_set(lc,L,M,add+addv[o]);
 59         clear_set(rc,M+1,R,add+addv[o]);
 60     } return;
 61 }
 62 inline int read(){
 63     int x=0,sig=1;char ch=getchar();
 64     while(!isdigit(ch)){if(ch==‘-‘)sig=-1;ch=getchar();}
 65     while(isdigit(ch))x=10*x+ch-‘0‘,ch=getchar();
 66     return x*=sig;
 67 }
 68 inline void write(int x){
 69     if(x==0){putchar(‘0‘);return;}if(x<0)putchar(‘-‘),x=-x;
 70     int len=0,buf[15];while(x)buf[len++]=x%10,x/=10;
 71     for(int i=len-1;i>=0;i--)putchar(buf[i]+‘0‘);return;
 72 }
 73 void init(){
 74     n=read();
 75     for(int i=1;i<n;i++){
 76         int a=read(),b=read();
 77         AddEdge(a,b,0);AddEdge(b,a,0);
 78         e[i]=(Edge){a,b,0};
 79     }
 80     dfs(1);build(1,1);
 81     for(int i=1;i<n;i++){
 82         if(dep[e[i].from]>dep[e[i].to]) swap(e[i].from,e[i].to);
 83     }
 84     return;
 85 }
 86 void work(){
 87     Q=read();
 88     while(Q--){
 89         int a=read(),b=read(),c=read();
 90         cv=c;change(a,b);
 91     }
 92     return;
 93 }
 94 void print(){
 95     clear_set(1,1,n,0);
 96     for(int i=1;i<=n;i++){
 97         write(sumv[num[i]]);ENT;
 98     }
 99     return;
100 }
101 int main(){init();work();print();return 0;}