【题解】[ZJOI2008] 树的统计

树链剖分板子

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define ls id<<1
#define rs id<<1|1
#define FOR(i,n,m) for(int i=n;i<=m;++i)
using namespace std;

const int N=100010;
int n,q,cnt,tot,sum,maxx;
int val[N],size[N],fa[N],dep[N],son[N],head[N];
struct EDGE { int to,next; } e[N];
int idx[N],rev[N],top[N];//rev[idx[x]]=x
struct node { int maxx,sum; } a[N];

inline int read()
{
    int x=0,f=1; char ch=getchar();
    while(ch<'0'||ch>'9') {if(ch=='-') f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9') x=(x<<3)+(x<<1)+ch-'0',ch=getchar();
    return x*f;
}

void add(int x,int y)
{
    e[++cnt]=(EDGE){y,head[x]};
    head[x]=cnt;
}

#define QXX(u) for(int i=head[u],v;v=e[i].to,i;i=e[i].next)

void dfs1(int u)
{
	size[u]=1;
	QXX(u) if(v!=fa[u]) {
		dep[v]=dep[u]+1;
		fa[v]=u;
		dfs1(v);
		size[u]+=size[v];
		if(size[v]>size[son[u]]) son[u]=v;
	}
}

void dfs2(int u)
{
	if(son[u])
	{
		top[son[u]]=top[u];
		idx[son[u]]=++tot;
		rev[tot]=son[u];
		dfs2(son[u]);
	}
	QXX(u) if(v!=fa[u]&&v!=son[u]) {
		top[v]=v;
		idx[v]=++tot;
		rev[tot]=v;
		dfs2(v);
	}
}

void update(int id)
{
	a[id].sum=a[ls].sum+a[rs].sum;
	a[id].maxx=max(a[ls].maxx,a[rs].maxx);
}

void build(int id,int l,int r)
{
	if(l==r)
	{
		a[id].maxx=a[id].sum=val[rev[l]];
		return ;
	}
	int mid=(l+r)>>1;
	build(ls,l,mid);
	build(rs,mid+1,r);
	update(id);
}

void revise(int id,int l,int r,int k,int v)
{
	if(l>k||r<k) return;
    if(l==r&&l==k) {
        a[id].sum=a[id].maxx=v;
        return;
    }
    int mid=(l+r)>>1; 
    if(mid>=k) revise(ls,l,mid,k,v);
    else revise(rs,mid+1,r,k,v);
    update(id);
}

void query(int id,int l,int r,int x,int y) 
{
    if(x>r||y<l) return;
    if(l>=x&&r<=y) 
	{
        sum+=a[id].sum;
        maxx=max(maxx,a[id].maxx);
        return;
    }
    int mid=(l+r)>>1;
    if(mid>=x) query(ls,l,mid,x,y);
    if(mid<y) query(rs,mid+1,r,x,y);
}

void ask(int x,int y)
{
	int tx=top[x],ty=top[y];
    while(tx!=ty) {
        if(dep[tx]<dep[ty]) swap(tx,ty),swap(x,y);
        query(1,1,tot,idx[tx],idx[x]);
        x=fa[tx],tx=top[x];
    }
    if(dep[x]>dep[y]) swap(x,y);
    query(1,1,tot,idx[x],idx[y]);
}

int main()
{
	n=read();
    FOR(i,1,n-1)
    {
        int x=read(),y=read();
        add(x,y); add(y,x);
    }
    FOR(i,1,n) val[i]=read();
    dfs1(1);
    tot=idx[1]=top[1]=rev[1]=1;
    dfs2(1);
    build(1,1,n);
    q=read();
    while(q--)
    {
        string s; cin>>s;
        int x=read(),y=read();
        if(s=="CHANGE") revise(1,1,tot,idx[x],y);
        else {
            maxx=-0x3f3f3f3f; sum=0;
            ask(x,y);
            if(s=="QMAX") printf("%d\n",maxx);
            else printf("%d\n",sum);
        }
    }
    return 0;
}