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 10000005using 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));        }    }}