[ZJOI2008] [BZOJ1036] 树的统计Count

这个就不多说了，树链剖分模板题。

#include "stdio.h"
#include "math.h"
#include "iostream"
#include "string.h"
#include "vector"
using namespace std;
const int N=30005,Lg=14,inf=(int)(2e9);
int n,tmp;
int fa[N][Lg+1];//fa[i][j]表示i向上2^j的祖先 
int v[N];//v[i]是第i个点的权值 
struct seg{ int l,r,sum,mx;}tr[N<<2];
vector <int> e[N];//边集数组 
int size[N];//size[i]表示大小 
bool vis[N];//是否访问 
int deep[N];//深度 
int belong[N];//相当于重链编号，当前重链的根节点 
int pos[N];//当前点对应的线段树编号 
void init(){
     cin>>n;
     int i,a,b;
     for (i=1;i<n;i++){
         scanf("%d%d",&a,&b);
         e[a].push_back(b);
         e[b].push_back(a);
     }
     for(i=1;i<=n;i++) 
      scanf("%d",v+i);
}
void dfs1(int root){
     int i;
     for (i=1;i<=Lg;i++){
         if(deep[root]<(1<<i)) break;
         fa[root][i]=fa[fa[root][i-1]][i-1];//倍增 
     }
     for (i=0;i<e[root].size();i++){
         if(vis[e[root][i]]==true) continue;
         deep[e[root][i]]=deep[root]+1;
         fa[e[root][i]][0]=root;
         vis[e[root][i]]=true;
         deep[e[root][i]]=deep[root]+1;
         dfs1(e[root][i]);
         vis[e[root][i]]=false;
         size[root]+=size[e[root][i]];
     }
     size[root]++;
}
void dfs2(int root,int bel){
     int i,k=-1,siz=0;
     belong[root]=bel;
     pos[root]=++tmp;
     for (i=0;i<e[root].size();i++){
         if(size[e[root][i]]>siz&&deep[root]<deep[e[root][i]])
          k=e[root][i],siz=size[e[root][i]];
     }
     if(k==-1) return ;
     dfs2(k,bel);
     for (i=0;i<e[root].size();i++){
         if(k!=e[root][i]&&deep[root]<deep[e[root][i]])
          dfs2(e[root][i],e[root][i]);
     }
}
void build(int l,int r,int tmp){
     tr[tmp].l=l,tr[tmp].r=r;
     if(l==r) return;
     build(l,(l+r)>>1,tmp<<1);
     build((l+r+2)>>1,r,tmp<<1|1);
}
void change(int posi,int name,int cto){
     if(tr[posi].l==tr[posi].r)
     {tr[posi].sum=tr[posi].mx=cto; return;}
     int mid=(tr[posi].l+tr[posi].r)>>1;
     if(name<=mid) change(posi<<1,name,cto);
     else change(posi<<1|1,name,cto);
     tr[posi].mx=max(tr[posi<<1].mx,tr[posi<<1|1].mx);
     tr[posi].sum=tr[posi<<1].sum+tr[posi<<1|1].sum;
}
int querymax(int p,int l,int r){
    if(tr[p].l==l&&tr[p].r==r) return tr[p].mx;
    int mid=(tr[p].l+tr[p].r)>>1;
    if(r<=mid) return querymax(p<<1,l,r);
    if(l>mid) return querymax(p<<1|1,l,r);
    return max(querymax(p<<1,l,mid),querymax(p<<1|1,mid+1,r));
}
int querysum(int p,int l,int r){
    if(tr[p].l==l&&tr[p].r==r) return tr[p].sum;
    int mid=(tr[p].l+tr[p].r)>>1;
    if(r<=mid) return querysum(p<<1,l,r);
    if(l>mid) return querysum(p<<1|1,l,r);
    return querysum(p<<1,l,mid)+querysum(p<<1|1,mid+1,r);
}
int lca(int x,int y){
    if(deep[x]<deep[y]) x+=y,y=x-y,x-=y;
    int len=deep[x]-deep[y],i;
    for (i=Lg;i>=0;i--)
        if(len&(1<<i)) x=fa[x][i];
    for (i=Lg;i>=0;i--)
        if(fa[x][i]!=fa[y][i]) x=fa[x][i],y=fa[y][i];
    if(x==y) return x;
    return fa[x][0];
}
int qmax(int u,int v){
    int mx=-inf;
    while (belong[u]!=belong[v]){
          mx=max(mx,querymax(1,pos[belong[u]],pos[u]));
          u=fa[belong[u]][0];
    }
    mx=max(mx,querymax(1,pos[v],pos[u]));
    return mx;
}
int qsum(int u,int v){
    int sum=0;
    while (belong[u]!=belong[v]){
          sum+=querysum(1,pos[belong[u]],pos[u]);
          u=fa[belong[u]][0];
    }
    sum+=querysum(1,pos[v],pos[u]);
    return sum;
}
void work(){
     int i,q,a,b;
     char s[10];
     build(1,n,1);
     for (i=1;i<=n;i++) change(1,pos[i],v[i]);
     scanf("%d",&q);
     for (i=1;i<=q;i++){
         scanf("%s%d%d",s,&a,&b);
         if(s[0]=='C') change(1,pos[a],b),v[a]=b;
         else{ int t=lca(a,b);
               if(s[1]=='M') printf("%d\n",max(qmax(a,t),qmax(b,t)));
               else printf("%d\n",qsum(a,t)+qsum(b,t)-v[t]);
         }
     }
}
int main(){
    init();
    vis[1]=true;
    dfs1(1);
    dfs2(1,1);
    vis[1]=false;
    work();
    return 0;
}