【洛谷P2590】树的统计【树链剖分】

$link$

分析：

嗯树链剖分模板
不过模板好像比这题更复杂？这题单点修改区间查询就行了

CODE：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
#include<algorithm>
using namespace std;
const int N=5e4+5,inf=0x7fffffff;
struct node
{
    int to,next;
}edge[N<<1];
int head[N],tot,num,dfn[N],son[N],fa[N];
int top[N],size[N],dep[N],n,id[N],k[N],T;
struct qaq
{
    int val,Max;
}a[N<<2];
void up(int x)
{
	a[x].val=a[x<<1].val+a[x<<1|1].val;
	a[x].Max=max(a[x<<1].Max,a[x<<1|1].Max);
}
void add(int x,int y)
{
    edge[++tot]=(node){y,head[x]};
    head[x]=tot;
}
void build(int x,int l,int r)
{
    if(l==r)
	{
		a[x].val=a[x].Max=k[id[l]];
		return;
	}
    int mid=(l+r)>>1;
    build(x<<1,l,mid);
	build(x<<1|1,mid+1,r);
    up(x);
}
void update(int x,int l,int r,int p,int k)
{
    if(l==r)
	{
		a[x].val=a[x].Max=k;
		return;
	}
    int mid=(l+r)>>1;
    if(p<=mid) update(x<<1,l,mid,p,k);
    else update(x<<1|1,mid+1,r,p,k);
    up(x);
}
int SumQ(int x,int l,int r,int L,int R)
{
    if(l==L&&r==R) return a[x].val;
    int mid=(l+r)>>1;
    if(R<=mid) return SumQ(x<<1,l,mid,L,R);
    if(L>mid) return SumQ(x<<1|1,mid+1,r,L,R);
    return SumQ(x<<1,l,mid,L,mid)+SumQ(x<<1|1,mid+1,r,mid+1,R);
}
int MaxQ(int x,int l,int r,int L,int R)
{
    if(l==L&&r==R) return a[x].Max;
    int mid=(l+r)>>1;
    if(R<=mid)return MaxQ(x<<1,l,mid,L,R);
    if(L>mid)return MaxQ(x<<1|1,mid+1,r,L,R);
    return max(MaxQ(x<<1,l,mid,L,mid),MaxQ(x<<1|1,mid+1,r,mid+1,R));
}
int queryS(int x,int y)
{
    int res=0;
    while(top[x]!=top[y])
    {
        if(dep[top[x]]<dep[top[y]])
            swap(x,y);
        res+=SumQ(1,1,n,dfn[top[x]],dfn[x]);
        x=fa[top[x]];
    }
    if(dep[x]>dep[y]) swap(x,y);
    res+=SumQ(1,1,n,dfn[x],dfn[y]);
    return res;
}
int queryM(int x,int y)
{
    int ans=-inf;
    while(top[x]!=top[y])
    {
        if(dep[top[x]]<dep[top[y]])
            swap(x,y);
        ans=max(ans,MaxQ(1,1,n,dfn[top[x]],dfn[x]));
        x=fa[top[x]];
    }
    if(dep[x]>dep[y]) swap(x,y);
    ans=max(ans,MaxQ(1,1,n,dfn[x],dfn[y]));
    return ans;
}
void dfs(int x,int father,int deep)
{
    dep[x]=deep;
	size[x]=1;
	fa[x]=father;
    for(int i=head[x];i;i=edge[i].next)
    {
        int qwq=edge[i].to;
        if(qwq==father)continue;
        dfs(qwq,x,deep+1);
        if(size[qwq]>size[son[x]]) son[x]=qwq;
        size[x]+=size[qwq];
    }
}
void dfs2(int x,int topN)
{
    dfn[x]=++num;
	id[num]=x;
	top[x]=topN;
    if(!son[x]) return;
	dfs2(son[x],topN);
    for(int i=head[x];i;i=edge[i].next)
    {
        int qwq=edge[i].to;
        if(qwq==fa[x]||qwq==son[x]) continue;
        dfs2(qwq,qwq);
    }
}
int main()
{
    scanf("%d",&n);
    for(int i=1,x,y;i<n;i++)
    {
    	scanf("%d%d",&x,&y);
        add(x,y);add(y,x);
    }
    dfs(1,0,1);
	dfs2(1,1);
    for(int i=1;i<=n;i++)
    	scanf("%d",&k[i]);
    build(1,1,n);
    scanf("%d",&T);
    while(T--)
    {
        char op[8];
        scanf("%s",op);
        int x,y;
        scanf("%d%d",&x,&y);
        if(op[0]=='C')
		{
			update(1,1,n,dfn[x],y);
			continue;
		}
        else
        {
        	if(op[1]=='S')
        	{
        		printf("%d\n",queryS(x,y));
        		continue;
			}
            else if(op[1]=='M')
			{
				printf("%d\n",queryM(x,y));
				continue;
			} 
		}
    }
    return 0;
}