bzoj 1036 [ZJOI2008]树的统计Count 树链剖分模板

  1 #include<stdio.h>
  2 #include<string.h>
  3 #include<stdlib.h>
  4 #include<iostream>
  5 #define maxn 233333
  6 #define lson ro<<1
  7 #define rson ro<<1|1
  8 #define inf 1e9
  9 using namespace std;
 10 struct node{int to,next;};
 11 struct rode{int l,r,sum,mx;};
 12 node e[maxn];
 13 rode tr[maxn];
 14 int n,v[30010],pos[30010],dep[30010],pre[maxn],cnt,fa[30010],size[30010],sz,bl[30010],mxx,an,m;
 15 int read();
 16 void insert(int,int);
 17 void build(int,int,int);
 18 void pushup(int);
 19 void dfs1(int);
 20 void dfs2(int,int);
 21 void querymx(int,int,int);
 22 void querysum(int,int,int);
 23 int solvemx(int,int);
 24 int solvesum(int,int);
 25 void change(int,int,int);
 26 int main(){
 27     n=read();
 28     for(int i=1;i<n;i++)insert(read(),read());
 29     for(int i=1;i<=n;i++)v[i]=read();
 30     dfs1(1);
 31     dfs2(1,1);
 32     build(1,n,1);
 33     for(int i=1;i<=n;i++) change(pos[i],v[i],1);
 34     m=read();
 35     for(int i=1;i<=m;i++){
 36         char cuan[12];
 37         scanf(" %s",cuan);
 38         if(cuan[1]==‘H‘){
 39             int x=read(),k=read();
 40             change(pos[x],k,1);
 41         }
 42         else if(cuan[1]==‘M‘) printf("%d\n",solvemx(read(),read()));
 43         else printf("%d\n",solvesum(read(),read()));
 44     }
 45     return 0;
 46 }
 47 int read(){
 48     int ans=0,f=1;char c=getchar();
 49     while(‘0‘>c||c>‘9‘){if(c==‘-‘)f=-1;c=getchar();}
 50     while(‘0‘<=c&&c<=‘9‘)ans=ans*10+c-48,c=getchar();return ans*f;
 51 }
 52 void insert(int x,int y){
 53     e[++cnt].to=y;e[cnt].next=pre[x];pre[x]=cnt;
 54     e[++cnt].to=x;e[cnt].next=pre[y];pre[y]=cnt;
 55 }
 56 void build(int l,int r,int ro){
 57     tr[ro].l=l;tr[ro].r=r;
 58     if(l==r) return;
 59     int mid=(l+r)>>1;
 60     build(l,mid,lson);build(mid+1,r,rson);
 61 }
 62 void pushup(int ro){
 63     tr[ro].sum=tr[lson].sum+tr[rson].sum;
 64     tr[ro].mx=max(tr[lson].mx,tr[rson].mx);
 65 }
 66 void dfs1(int x){
 67     size[x]=1;
 68     for(int i=pre[x];i;i=e[i].next){
 69         int to=e[i].to;
 70         if(fa[x]==to)continue;
 71         fa[to]=x;
 72         dep[to]=dep[x]+1;
 73         dfs1(to);
 74         size[x]+=size[to];
 75     }
 76 }
 77 void dfs2(int x,int chain){
 78     int k=0;
 79     pos[x]=++sz;
 80     bl[x]=chain;
 81     for(int i=pre[x];i;i=e[i].next)
 82        if(dep[e[i].to]>dep[x]&&size[k]<size[e[i].to])
 83           k=e[i].to;
 84     if(k==0)return;
 85     dfs2(k,chain);
 86     for(int i=pre[x];i;i=e[i].next)
 87        if(dep[e[i].to]>dep[x]&&k!=e[i].to)
 88           dfs2(e[i].to,e[i].to);
 89 }
 90 void change(int x,int k,int ro){
 91     if(tr[ro].r==tr[ro].l){
 92         tr[ro].sum=tr[ro].mx=k;
 93         return;
 94     }
 95     int mid=(tr[ro].r+tr[ro].l)>>1;
 96     if(x<=mid) change(x,k,lson);
 97     else change(x,k,rson);
 98     pushup(ro);
 99 }
100 void querymx(int l,int r,int ro){
101     if(l<=tr[ro].l&&tr[ro].r<=r){
102         mxx=max(mxx,tr[ro].mx);
103         return;
104     }
105     int mid=(tr[ro].l+tr[ro].r)>>1;
106     if(l<=mid) querymx(l,r,lson);
107     if(r>mid)  querymx(l,r,rson);
108 }
109 void querysum(int l,int r,int ro){
110     if(l<=tr[ro].l&&tr[ro].r<=r){
111         an+=tr[ro].sum;
112         return;
113     }
114     int mid=(tr[ro].l+tr[ro].r)>>1;
115     if(l<=mid) querysum(l,r,lson);
116     if(r>mid)  querysum(l,r,rson);
117 }
118 int solvemx(int x,int y){
119     int mxs=-inf;
120     while(bl[x]!=bl[y]){
121         if(dep[bl[x]]<dep[bl[y]])swap(x,y);
122         mxx=-inf;querymx(pos[bl[x]],pos[x],1);
123         mxs=max(mxs,mxx);
124         x=fa[bl[x]];
125     }
126     if(pos[x]>pos[y])swap(x,y);
127     mxx=-inf;querymx(pos[x],pos[y],1);
128     mxs=max(mxs,mxx);
129     return mxs;
130 }
131 int solvesum(int x,int y){
132     int sum=0;
133     while(bl[x]!=bl[y]){
134         if(dep[bl[x]]<dep[bl[y]])swap(x,y);
135         an=0;querysum(pos[bl[x]],pos[x],1);
136         sum+=an;
137         x=fa[bl[x]];
138     }
139     if(pos[x]>pos[y])swap(x,y);
140     an=0;querysum(pos[x],pos[y],1);
141     sum+=an;
142     return sum;
143 }