BZOJ 3531 [Sdoi2014]旅行 树链剖分

//最近在做树剖专题，由于我比较弱，所以只能从水题开始。
传送门
大概意思就是他把树进行染色，每个节点有一个点权，有若干操作
1.将一个节点的颜色改变
2.将一个节点的点权改变
3.求两点间简单路径上颜色相同的点权的最大值
4.求两点间简单路径上颜色相同的点权和

内存限制512M
这给了我们以暴力的空间，直接把每一个颜色都开一个线段树，各个点在线段树上的位置是一样的，但在不同线段树上的值不同。
为了节省空间，不妨动态开点，要不然就MLE滚粗吧。
剩下的就是代码能力了，最好不要调试一遍过，这题的数据最小的12K，1000个点，错一个数给你你都不知道怎么调

对了，在线段树上的修改之后别忘了在原数组上进行修改
下面是代码,我写的既长又恶心，仅供参考

#include <stdio.h>
#include <string.h>
#include <algorithm>
#include <iostream>
#define N 100003
#define M 10000005
using namespace std;
int head[N],top[N],w[N],size[N];
int depth[N],fa[N],sz,tree,root[N],n,m;
int a[N],b[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<<1)+(x<<3)+ch-'0';ch = getchar();}
    return x*f;
}
struct graph
{
    int next,to;
    graph () {}
    graph (int _next,int _to)
    :next(_next),to(_to){}
}edge[M];
inline void add(int x,int y)
{
    static int cnt = 0;
    edge[++cnt] = graph(head[x],y);
    head[x]=cnt;
    edge[++cnt] = graph(head[y],x);
    head[y]=cnt;
}
void DFS1(int x)
{
    size[x]=1;
    for(int i=head[x];i;i=edge[i].next)
        if(edge[i].to!=fa[x])
        {
            depth[edge[i].to]=depth[x]+1;
            fa[edge[i].to]=x;
            DFS1(edge[i].to);
            size[x]+=size[edge[i].to];
        }
}
void DFS2(int x,int chain)
{
    top[x]=chain;
    w[x]=++sz;
    int k = 0;
    for(int i=head[x];i;i=edge[i].next)
        if(edge[i].to!=fa[x]&&size[k]<size[edge[i].to])
            k=edge[i].to;
    if(!k)return;
    DFS2(k,chain);
    for(int i=head[x];i;i=edge[i].next)
        if(edge[i].to!=fa[x]&&edge[i].to!=k)
            DFS2(edge[i].to,edge[i].to);
}
struct seg
{
    int lson,rson;
    int mx,mn;
}tr[M];
inline void updata(int k)
{
    tr[k].mn = tr[tr[k].lson].mn + tr[tr[k].rson].mn;
    tr[k].mx = max(tr[tr[k].rson].mx,tr[tr[k].lson].mx);
}
void change(int &k,int l,int r,int x,int v)
{
    if(!k)k=++tree;
    if(l==r){tr[k].mn = tr[k].mx = v;return;}
    int mid = (l+r)>>1;
    if(x<=mid)change(tr[k].lson,l,mid,x,v);
    else change(tr[k].rson,mid+1,r,x,v);
    updata(k);
}
int asksum(int k,int l,int r,int x,int y)
{
    if(!k)return 0;
    if(l==x && r==y)return tr[k].mn;
    int mid = (l+r)>>1;
    if(y<=mid)return asksum(tr[k].lson,l,mid,x,y);
    else if(x>mid)return asksum(tr[k].rson,mid+1,r,x,y);
    else return asksum(tr[k].lson,l,mid,x,mid)+asksum(tr[k].rson,mid+1,r,mid+1,y);
}
int askmax(int k,int l,int r,int x,int y)
{
    if(!k)return 0;
    if(l==x && r==y)return tr[k].mx;
    int mid = (l+r)>>1;
    if(y<=mid)return askmax(tr[k].lson,l,mid,x,y);
    else if(x>mid)return askmax(tr[k].rson,mid+1,r,x,y);
    else return max(askmax(tr[k].lson,l,mid,x,mid),askmax(tr[k].rson,mid+1,r,mid+1,y));
}
int lca(int x,int y)
{
    while(top[x]^top[y])
    {
        if(depth[top[x]]<depth[top[y]])
            swap(x,y);
        x=fa[top[x]];
    }
    return depth[x]<depth[y]?x:y;
}
int solvemax(int k,int x,int y)
{
    int MAX = 0;
    while(top[x]!=top[y])
    {
        if(depth[top[x]]<depth[top[y]])swap(x,y);
        MAX = max(MAX, askmax(k,1,n,w[top[x]],w[x]) );
        x=fa[top[x]];
    }
    if(depth[x]>depth[y])swap(x,y);
    MAX = max(MAX,askmax(k,1,n,w[x],w[y]));
    return MAX;
}
int solvesum(int k,int x,int y)
{
    int sum = 0;
    while(top[x]!=top[y])
    {
        if(depth[top[x]]<depth[top[y]])swap(x,y);
        sum += asksum(k,1,n,w[top[x]],w[x]);
        x=fa[top[x]];
    }
    if(depth[x]>depth[y])swap(x,y);
    sum += asksum(k,1,n,w[x],w[y]);
    return sum;
}
int main()
{
    n=read(),m=read();
    for(int i=1;i<=n;++i)a[i]=read(),b[i]=read();
    for(int i=1;i<n;++i)
    {
        int x=read(),y=read();
        add(x,y);
    }
    depth[1]=1;
    DFS1(1);
    DFS2(1,1);
    for(int i=1;i<=n;++i)
        change(root[b[i]],1,n,w[i],a[i]);
    for(int i=1;i<=m;++i)
    {
        char str[40];scanf("%s",str);
        int x=read(),y=read();
        if(str[1]=='C')
        {
            change(root[b[x]],1,n,w[x],0);
            b[x]=y;
            change(root[b[x]],1,n,w[x],a[x]);
        }
        else if(str[1]=='S')
        {
            printf("%d\n",solvesum(root[b[x]],x,y));
        }
        else if(str[1]=='W')
        {
            change(root[b[x]],1,n,w[x],y);
            a[x]=y;
        }
        else
        {
            printf("%d\n",solvemax(root[b[x]],x,y));
        }
    }
}