【BZOJ3531】旅行，树链剖分+开点线段树

传送门
写在前面：为自己和并肩作战的两个队友默哀
思路：感觉做的链剖题越来越厉害了，这道题让我接触到了线段树的一种新姿势——原来写线段树，对于某个节点i，左右儿子是直接当成$i*2$和$i*2+1$来搞的，但是发现对于这道题貌似并不适用，所以还是一步步编号的更好，记录左右儿子的编号，这样做就可以避免编号重复，而且可以充分利用空间建出多个线段树了。还有就是这道题是要对线段树内的点进行动态删除与增加，对于这个问题，由于我们只需要记录的有效信息就是最大值与总和，所以删除某个节点就可以把这个点权值修改为0，添加点就比较简单，直接把权值扔进去开点编号即可，其他操作和普通链剖没啥区别
注意：线段树的数组一定要开大点，因为可能会有很多改动操作……
代码：

#include<bits/stdc++.h>
using namespace std;
int n,m,cnt,tot,al;
char ch;
int roots[100010],pre[100010],c[100010],w[100010],top[100010],fa[100010],son[100010],siz[100010],dep[100010],L[100010],first[100010];
struct os
{
    int u,v,next;
}e[200010];
struct node
{
    int maxn,sum,lson,rson;
}tree[4000010];
void pushup(int now)
{
    tree[now].maxn=max(tree[tree[now].lson].maxn,tree[tree[now].rson].maxn);
    tree[now].sum=tree[tree[now].lson].sum+tree[tree[now].rson].sum;
}
void add(int x,int y)
{
    e[++tot].u=x;
    e[tot].v=y;
    e[tot].next=first[x];
    first[x]=tot;
}
void dfs1(int now)
{
    siz[now]=1;
    for (int i=first[now];i;i=e[i].next)
    if (e[i].v!=fa[now])
    {
        fa[e[i].v]=now;
        dep[e[i].v]=dep[now]+1;
        dfs1(e[i].v);
        if (siz[e[i].v]>siz[son[now]]) son[now]=e[i].v;
        siz[now]+=siz[e[i].v];
    }
}
void dfs2(int now,int tp)
{
    top[now]=tp;
    pre[++cnt]=now;
    L[now]=cnt;
    if (son[now]) dfs2(son[now],tp);
    for (int i=first[now];i;i=e[i].next)
    if (e[i].v!=fa[now]&&e[i].v!=son[now]) dfs2(e[i].v,e[i].v);
}
void update(int &now,int begin,int end,int pos,int num)
{
    if (!now) now=++al;
    if (begin==end) {tree[now].maxn=tree[now].sum=num;return;}
    int mid=(begin+end)>>1;
    if (pos<=mid) update(tree[now].lson,begin,mid,pos,num);
    else update(tree[now].rson,mid+1,end,pos,num);
    pushup(now);
}
int get_sum(int now,int begin,int end,int l,int r)
{
    if (l<=begin&&end<=r) return tree[now].sum;
    int mid=(begin+end)>>1,ans=0;
    if (mid>=l) ans+=get_sum(tree[now].lson,begin,mid,l,r);
    if (mid<r) ans+=get_sum(tree[now].rson,mid+1,end,l,r);
    return ans;
}
int get_max(int now,int begin,int end,int l,int r)
{
    if (l<=begin&&end<=r) return tree[now].maxn;
    int mid=(begin+end)>>1,ans=0;
    if (mid>=l) ans=max(ans,get_max(tree[now].lson,begin,mid,l,r));
    if (mid<r) ans=max(ans,get_max(tree[now].rson,mid+1,end,l,r));
    return ans;
}
void solve(int color,int l,int r)
{
    int f1=top[l],f2=top[r],ans=0;
    while (f1!=f2)
    {
        if (dep[f1]<dep[f2]) swap(f1,f2),swap(l,r);
        if (ch=='S') ans+=get_sum(roots[color],1,cnt,L[f1],L[l]);
        else ans=max(ans,get_max(roots[color],1,cnt,L[f1],L[l]));
        l=fa[f1];f1=top[l];
    }
    if (dep[l]>dep[r]) swap(l,r);
    if (ch=='S') ans+=get_sum(roots[color],1,cnt,L[l],L[r]);
    else ans=max(ans,get_max(roots[color],1,cnt,L[l],L[r]));
    printf("%d\n",ans);
}
main()
{
    scanf("%d%d",&n,&m);
    int x,y,z;
    for (int i=1;i<=n;i++)
    scanf("%d%d",&w[i],&c[i]);
    for (int i=1;i<n;i++)
    scanf("%d%d",&x,&y),
    add(x,y),add(y,x);
    dfs1(1);
    dfs2(1,1);
    for (int i=1;i<=n;i++)
    update(roots[c[i]],1,cnt,L[i],w[i]);
    while (m--)
    {
        ch=getchar();
        while (ch!='C'&&ch!='Q') ch=getchar();
        if (ch=='C')
        {
            ch=getchar();
            scanf("%d%d",&x,&y);
            if (ch=='C')
                update(roots[c[x]],1,n,L[x],0),
                c[x]=y,
                update(roots[c[x]],1,n,L[x],w[x]);
            else
                update(roots[c[x]],1,n,L[x],y),w[x]=y;
        }
        else
            ch=getchar(),
            scanf("%d%d",&x,&y),
            solve(c[x],x,y);
    }
}