洛谷-3313 [SDOI2014]旅行

题目描述
S国有N个城市，编号从1到N。城市间用N-1条双向道路连接，满足从一个城市出发可以到达其它所有城市。每个城市信仰不同的宗教，如飞天面条神教、隐形独角兽教、绝地教都是常见的信仰。
为了方便，我们用不同的正整数代表各种宗教， S国的居民常常旅行。旅行时他们总会走最短路，并且为了避免麻烦，只在信仰和他们相同的城市留宿。当然旅程的终点也是信仰与他相同的城市。S国政府为每个城市标定了不同的旅行评级，旅行者们常会记下途中（包括起点和终点）留宿过的城市的评级总和或最大值。
在S国的历史上常会发生以下几种事件：
“CC x c“：城市x的居民全体改信了c教；
“CW x w“：城市x的评级调整为w;
“QS x y“：一位旅行者从城市x出发，到城市y，并记下了途中留宿过的城市的评级总和；
“QM x y“：一位旅行者从城市x出发，到城市y，并记下了途中留宿过的城市的评级最大值。
由于年代久远，旅行者记下的数字已经遗失了，但记录开始之前每座城市的信仰与评级，还有事件记录本身是完好的。请根据这些信息，还原旅行者记下的数字。 为了方便，我们认为事件之间的间隔足够长，以致在任意一次旅行中，所有城市的评级和信仰保持不变。
输入格式
输入的第一行包含整数N，Q依次表示城市数和事件数。
接下来N行，第i+l行两个整数Wi，Ci依次表示记录开始之前，城市i的评级和信仰。 接下来N-1行每行两个整数x，y表示一条双向道路。
接下来Q行，每行一个操作，格式如上所述。
输出格式
对每个QS和QM事件，输出一行，表示旅行者记下的数字。

输入输出样例
输入 #1
5 6
3 1
2 3
1 2
3 3
5 1
1 2
1 3
3 4
3 5
QS 1 5
CC 3 1
QS 1 5
CW 3 3
QS 1 5
QM 2 4

输出 #1
8
9
11
3

说明/提示
$N，Q<=10^5，C<=10^5$

解释：树链剖分直接上模板，不同宗教不同线段树来维护，同时我们进行动态开点，就可以保证通过

#include <cstdio>
#include <iostream>
#define il inline
const int maxm=1e6+1;
struct tree{
    int ls,rs,maxx,sum;
};
int root[maxm],rt;
tree st[maxm*4];
int deep[maxm],son[maxm],fa[maxm],size[maxm],top[maxm];
int head[maxm],to[maxm*2],net[maxm*2],cnt;
int yval[maxm],val[maxm],yc[maxm],c[maxm];
int xz[maxm];
int id[maxm],who[maxm],tot;
void add_edge(int x,int y){
    to[++cnt]=y,net[cnt]=head[x],head[x]=cnt;
}
int dfs1(int x,int fax,int dep){
    fa[x]=fax,deep[x]=dep,size[x]=1;
    int maxson=-1;
    for(int i=head[x];i;i=net[i])
    if(to[i]!=fax){
        size[x]+=dfs1(to[i],x,dep+1);
        if(maxson<size[to[i]]) maxson=size[to[i]],son[x]=to[i];
    }
    return size[x];
}
void dfs2(int x,int topx){
    id[x]=++tot;
    who[tot]=x;
    val[tot]=yval[x],c[tot]=yc[x];
    top[x]=topx;
    if(!son[x]) return;
    dfs2(son[x],topx);
    for(int i=head[x];i;i=net[i])
     if(!id[to[i]]) dfs2(to[i],to[i]);
}
void update(int o){
    st[o].maxx=std::max(st[st[o].ls].maxx,st[st[o].rs].maxx);
    st[o].sum=st[st[o].ls].sum+st[st[o].rs].sum;
}
void mofidy(int &o,int l,int r,int ind,int num){
    if(!o) o=++rt;
    if(l>=r){
        st[o].maxx=st[o].sum=num;
        return;
    }
    int mid=(l+r)>>1;
    if(ind<=mid) mofidy(st[o].ls,l,mid,ind,num);
    else mofidy(st[o].rs,mid+1,r,ind,num);
    update(o);
}
int askmax(int o,int l,int r,int ql,int qr){
    if(!o) return -1;
    if(r<ql||l>qr) return -1;
    if(ql<=l&&r<=qr)
     return st[o].maxx;
    int mid=(l+r)>>1;
    return std::max(askmax(st[o].ls,l,mid,ql,qr),askmax(st[o].rs,mid+1,r,ql,qr));
}
int asksum(int o,int l,int r,int ql,int qr){
    if(!o) return 0;
    if(r<ql||l>qr) return 0;
    if(ql<=l&&r<=qr)
     return st[o].sum;
    int mid=(l+r)>>1;
    return asksum(st[o].ls,l,mid,ql,qr)+asksum(st[o].rs,mid+1,r,ql,qr);
}
int tree_max(int u,int v){
    int ans=-1;
    int xz=yc[v];
    while(top[u]!=top[v]){
        if(deep[top[u]]<deep[top[v]]) std::swap(u,v);
        ans=std::max(ans,askmax(root[xz],1,tot,id[top[u]],id[u]));
        u=fa[top[u]];
    }
    if(deep[u]<deep[v]) std::swap(u,v);
    ans=std::max(ans,askmax(root[xz],1,tot,id[v],id[u]));
    return ans;
}
int tree_sum(int u,int v){
    int ans=0;
    int xz=yc[v];
    while(top[u]!=top[v]){
        if(deep[top[u]]<deep[top[v]]) std::swap(u,v);
        ans+=asksum(root[xz],1,tot,id[top[u]],id[u]);
        u=fa[top[u]];
    }
    if(deep[u]<deep[v]) std::swap(u,v);
    ans+=asksum(root[xz],1,tot,id[v],id[u]);
    return ans;
}
int read(){
    int x=0,w=1;char ch=0;
    while(ch<'0'||ch>'9'){
        if(ch=='-') w=-1;
        ch=getchar();
    }
    while(ch>='0'&&ch<='9') x=(x<<3)+(x<<1)+ch-'0',ch=getchar();
    return x*w;
}
int main(){
    int n,q;
    n=read(),q=read();
    for(int i=1;i<=n;i++)
     yval[i]=read(),yc[i]=read();
    for(int i=1;i<n;i++){
        int u=read(),v=read();
        add_edge(u,v);
        add_edge(v,u);
    }
    dfs1(1,0,1),dfs2(1,1);
    for(int i=1;i<=n;i++)
     mofidy(root[yc[who[i]]],1,tot,i,yval[who[i]]);
    for(int i=1;i<=q;i++){
        char s[20];
        int x,y;
        scanf("%s",s);
        x=read(),y=read();
        if(s[0]=='C'&&s[1]=='C'){
            mofidy(root[yc[x]],1,tot,id[x],0);
            yc[x]=y;
            mofidy(root[yc[x]],1,tot,id[x],yval[x]);
        }
        if(s[0]=='C'&&s[1]=='W')
         yval[x]=y,mofidy(root[yc[x]],1,tot,id[x],y);
        if(s[0]=='Q'&&s[1]=='M')
         printf("%d\n",tree_max(x,y));
        if(s[0]=='Q'&&s[1]=='S')
         printf("%d\n",tree_sum(x,y));
    }
    return 0;
}