【ZJOI2008】树的统计

Description

给出一棵n个节点的树，和m个操作。每个点有点权，操作有单点修改，树上路径和和树上路径最大主三种。n<=30000,m<=200000,-30000<=权值<=30000。

Solution

裸的树链剖分。不多解释。
不会的请点击树链剖分学习小记

Code

#include<cstdio>
#include<cstring>
#include<algorithm>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define rep(i,a) for(int i=last[a];i;i=next[i])
#define N 30005
#define ll long long
using namespace std;
struct note{
    int max;ll sum;
}tree[N*5];
int t[N*2],next[N*2],last[N];
int n,m,l,tot,x,y,size[N],son[N],fa[N],top[N],w[N],d[N],ans;
ll sum;
char s[6];
void add(int x,int y) {
    t[++l]=y;next[l]=last[x];last[x]=l;
}
void dfs(int x,int y) {
    size[x]=1;fa[x]=y;d[x]=d[y]+1;int k=0;
    rep(i,x) if (t[i]!=y) {
        dfs(t[i],x);size[x]+=size[t[i]];
        if (size[t[i]]>k) k=size[t[i]],son[x]=t[i];
    }
}
void make(int x,int y) {
    top[x]=y;w[x]=++tot;
    if (!son[x]) return;
    make(son[x],y);
    rep(i,x) if (t[i]!=fa[x]&&t[i]!=son[x]) make(t[i],t[i]);
}
void change(int v,int l,int r,int x,int y) {
    if (l==r) {tree[v].sum=tree[v].max=y;return;}
    int m=(l+r)/2;
    if (x<=m) change(v*2,l,m,x,y);
    else change(v*2+1,m+1,r,x,y);
    tree[v].sum=tree[v*2].sum+tree[v*2+1].sum;
    tree[v].max=max(tree[v*2].max,tree[v*2+1].max);
}
void query(int v,int l,int r,int x,int y) {
    if (l==x&&r==y) {
        ans=max(ans,tree[v].max);sum+=tree[v].sum;return;
    }
    int m=(l+r)/2;
    if (y<=m) query(v*2,l,m,x,y);
    else if (x>m) query(v*2+1,m+1,r,x,y);
    else query(v*2,l,m,x,m),query(v*2+1,m+1,r,m+1,y);
}
void find(int x,int y) {
    int f1=top[x],f2=top[y];ans=-0x7fffffff,sum=0;
    while (f1!=f2) {
        if (d[f1]<d[f2]) swap(f1,f2),swap(x,y);
        query(1,1,n,w[f1],w[x]);x=fa[f1];f1=top[x];
    }
    if (d[x]>d[y]) swap(x,y);
    query(1,1,n,w[x],w[y]);
}
int main() {
    scanf("%d",&n);
    fo(i,1,n-1) scanf("%d%d",&x,&y),add(x,y),add(y,x);
    d[1]=1;dfs(1,0);make(1,1);
    fo(i,1,n) scanf("%d",&x),change(1,1,n,w[i],x);
    for(scanf("%d",&m);m;m--) {
        scanf("%s%d%d",s,&x,&y);
        if (s[0]=='C') change(1,1,n,w[x],y);
        else {
            find(x,y);
            if (s[1]=='M') printf("%d\n",ans);
            else printf("%lld\n",sum);
        }
    }
}