BZOJ 2243: [SDOI2011]染色 [树链剖分]

//
//  main.cpp
//  sdoi2011染色
//
//  Created by Candy on 2016/12/14.
//  Copyright © 2016年 Candy. All rights reserved.
//

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
#define lc o<<1
#define rc o<<1|1
#define m ((l+r)>>1)
#define lson o<<1,l,m
#define rson o<<1|1,m+1,r
const int N=1e5+5;
int read(){
    char c=getchar();int x=0,f=1;
    while(c<‘0‘||c>‘9‘){if(c==‘-‘)f=-1; c=getchar();}
    while(c>=‘0‘&&c<=‘9‘){x=x*10+c-‘0‘; c=getchar();}
    return x*f;
}
int n,Q,ww[N],w[N],x,y,a,b,c;
char s[10];
struct edge{
    int v,ne;
}e[N<<1];
int h[N],cnt;
inline void ins(int u,int v){
    cnt++;
    e[cnt].v=v;e[cnt].ne=h[u];h[u]=cnt;
    cnt++;
    e[cnt].v=u;e[cnt].ne=h[v];h[v]=cnt;
}
int top[N],size[N],tid[N],tot,fa[N],deep[N],mx[N];
void dfs(int u){
    size[u]++;
    for(int i=h[u];i;i=e[i].ne){
        int v=e[i].v;
        if(v==fa[u]) continue;
        fa[v]=u;deep[v]=deep[u]+1;
        dfs(v);
        size[u]+=size[v];
        if(size[v]>size[mx[u]]) mx[u]=v;
    }
}
void dfs(int u,int anc){
    if(!u) return;
    tid[u]=++tot;top[u]=anc;
    dfs(mx[u],anc);
    for(int i=h[u];i;i=e[i].ne)
        if(e[i].v!=fa[u]&&e[i].v!=mx[u]) dfs(e[i].v,e[i].v);
}

struct node{
    int lv,rv,cnt,tag;
}t[N<<2];
inline void merge(int o){
    t[o].tag=-1;
    t[o].lv=t[lc].lv;
    t[o].rv=t[rc].rv;
    t[o].cnt=t[lc].cnt+t[rc].cnt-(t[lc].rv==t[rc].lv);
}
inline void paint(int o,int v){
    t[o].tag=t[o].lv=t[o].rv=v;
    t[o].cnt=1;
}
void build(int o,int l,int r){
    if(l==r) paint(o,w[l]);
    else{
        build(lson);
        build(rson);
        merge(o);
    }
}
inline void pushDown(int o){
    if(t[o].tag!=-1){
        paint(lc,t[o].tag);
        paint(rc,t[o].tag);
        t[o].tag=-1;
    }
}
void change(int o,int l,int r,int ql,int qr,int v){
    if(ql<=l&&r<=qr) paint(o,v);
    else{
        pushDown(o);
        if(ql<=m) change(lson,ql,qr,v);
        if(m<qr) change(rson,ql,qr,v);
        merge(o);
    }
}
int query(int o,int l,int r,int ql,int qr){
    if(ql<=l&&r<=qr) return t[o].cnt;
    else if(t[o].tag!=-1) return 1;
    else{
        pushDown(o);
        int ans=0;
        if(ql<=m) ans+=query(lson,ql,qr);
        if(m<qr) ans+=query(rson,ql,qr);
        if(ql<=m&&m<qr&&t[lc].rv==t[rc].lv) ans--;
        return ans;
    }
}
int getcol(int o,int l,int r,int p){
    if(l==r) return t[o].lv;
    else if(t[o].tag!=-1) return t[o].tag;
    else{
        if(p<=m) return getcol(lson,p);
        else return getcol(rson,p);
    }
}
void sol1(int x,int y,int c){
    while(top[x]!=top[y]){
        if(deep[top[x]]<deep[top[y]]) swap(x,y);
        change(1,1,n,tid[top[x]],tid[x],c);
        x=fa[top[x]];
    }
    if(tid[x]>tid[y]) swap(x,y);
    change(1,1,n,tid[x],tid[y],c);
}
int sol2(int x,int y){
    int ans=0;
    while(top[x]!=top[y]){
        if(deep[top[x]]<deep[top[y]]) swap(x,y);
        ans+=query(1,1,n,tid[top[x]],tid[x]);
        if(getcol(1,1,n,tid[top[x]])==getcol(1,1,n,tid[fa[top[x]]])) ans--;
        x=fa[top[x]];
    }
    if(tid[x]>tid[y]) swap(x,y);
    ans+=query(1,1,n,tid[x],tid[y]);
    return ans;
}
int main(int argc, const char * argv[]) {
    n=read();Q=read();
    for(int i=1;i<=n;i++) ww[i]=read();
    for(int i=1;i<=n-1;i++) x=read(),y=read(),ins(x,y);
    dfs(1);dfs(1,1);
    for(int i=1;i<=n;i++) w[tid[i]]=ww[i];
    build(1,1,n);
    while(Q--){
        scanf("%s",s);a=read();b=read();
        if(s[0]==‘C‘){
            c=read();sol1(a,b,c);
        }else printf("%d\n",sol2(a,b));
    }
    return 0;
}