【Bzoj1036】树的统计

题意

$一棵树上有n个节点，编号分别为1到n，每个节点都有一个权值w。我们将以下面的形式来要求你对棵树完成一些操作：$
$1.把结点u的权值改为t$
$2.询问从点u到点v的路径上的节点的最大权值$
$3.询问从点u到点v的路径上的节点的权值和$
$注意：从点u到点v的路径上的节点包括u和v本身$

数据范围

n<=30000 ,t<=200000

---

树链剖分的模板，两个操作可以合并，减少编程难度。

#include <cstdio>
#include <algorithm>

using namespace std;

const int maxx = 30000 + 25;
const int Inf = (unsigned)(-1)>>1;

int head[maxx],to[maxx<<1],nxt[maxx<<1];
int a[maxx],dpt[maxx],rnk[maxx],size[maxx],ftr[maxx],top[maxx],son[maxx];
int T[maxx<<2],Tmax[maxx<<2];

int n,m,cnt,t,x,y,num;
char flag[8];

void Add(int x,int y){to[++num]=y;nxt[num]=head[x];head[x]=num;}
void chk(int pos,int &a,int b){
    if(pos == 0) a = a+b;
    if(pos == 1) a = max(a,b);
}

namespace Y{

    void Dfs1(int x){
        size[x] = 1;
        for(int i=head[x];i;i=nxt[i]){
            int now = to[i];if(now == ftr[x]) continue;
            ftr[now] = x;dpt[now] = dpt[x] + 1;
            Dfs1(now); size[x] += size[now];
            if(size[now] > size[son[x]]) son[x] = now;
        }
    }

    void Dfs2(int x,int brn){
        rnk[x] = ++cnt;top[x] = brn;
        if(son[x]) Dfs2(son[x],brn);
        for(int i=head[x];i;i=nxt[i])
            if(to[i] != ftr[x] && to[i] != son[x])
                Dfs2(to[i],to[i]);
    }

    void modify(int i,int l,int r,int pos,int k){
        if(l == r) {T[i] = Tmax[i] = k;return;}
        int mid = (l+r) >> 1;
        if(pos <= mid) modify(i<<1,l,mid,pos,k);
        if(pos >  mid) modify(i<<1|1,mid+1,r,pos,k);
        T[i]=T[i<<1]+T[i<<1|1];Tmax[i]=max(Tmax[i<<1],Tmax[i<<1|1]);
    }

    int Query(int i,int x,int y,int l,int r,int flag){
        int ans = flag? -Inf : 0;
        if(x<=l && r<=y) return flag? Tmax[i] : T[i];
        int mid = (l+r) >> 1;
        if(x <= mid) chk(flag,ans,Query(i<<1,x,y,l,mid,flag));
        if(y >  mid) chk(flag,ans,Query(i<<1|1,x,y,mid+1,r,flag));
        return ans;
    }

    int solve(int flag,int x,int y){
        int ans = flag? -Inf : 0;
        while(top[x] != top[y]){
            if(dpt[top[x]] > dpt[top[y]]) x^=y^=x^=y;
            chk(flag,ans,Query(1,rnk[top[y]],rnk[y],1,n,flag));
            y = ftr[top[y]];
        }
        if(rnk[x] > rnk[y]) x^=y^=x^=y;
        chk(flag,ans,Query(1,rnk[x],rnk[y],1,n,flag));
        return ans;
    }
}

using namespace Y;

int main(){
    scanf("%d",&n);
    for(int i=1;i<n;i++) scanf("%d%d",&x,&y),Add(x,y),Add(y,x);
    for(int i=1;i<=n;i++) scanf("%d",&a[i]);Dfs1(1);Dfs2(1,1);
    for(int i=1;i<=n;i++) modify(1,1,n,rnk[i],a[i]);
    scanf("%d",&t);
    while( t-- ){
        scanf("%s%d%d",flag,&x,&y);
        if(flag[1] == 'H') modify(1,1,n,rnk[x],y);
        if(flag[1] == 'M') printf("%d\n",solve(1,x,y));
        if(flag[1] == 'S') printf("%d\n",solve(0,x,y));
    }
    return 0;
}