【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;}