2243: [SDOI2011]染色 树链剖分+线段树染色

给定一棵有n个节点的无根树和m个操作，操作有2类：

1、将节点a到节点b路径上所有点都染成颜色c；

2、询问节点a到节点b路径上的颜色段数量（连续相同颜色被认为是同一段），

如“112221”由3段组成：“11”、“222”和“1”。

请你写一个程序依次完成这m个操作=。

题解：树链剖分+线段树，线段树维护区间左右的颜色和种类数，合并就很简单了，考虑左区间的右端点和右区间的左端点是不是一样颜色，然后树上暴跳的时候要比较深度来跳了，然后修改也是在树上暴跳即可

/**************************************************************
    Problem: 2243
    User: walfy
    Language: C++
    Result: Accepted
    Time:5192 ms
    Memory:48268 kb
****************************************************************/

//#pragma comment(linker, "/stack:200000000")
//#pragma GCC optimize("Ofast,no-stack-protector")
//#pragma GCC target("sse,sse2,sse3,ssse3,sse4,popcnt,abm,mmx,avx,tune=native")
//#pragma GCC optimize("unroll-loops")
#include<bits/stdc++.h>
#define fi first
#define se second
#define mp make_pair
#define pb push_back
#define pi acos(-1.0)
#define ll long long
#define vi vector<int>
#define mod 1000000007
#define C 0.5772156649
#define ls l,m,rt<<1
#define rs m+1,r,rt<<1|1
#define pil pair<int,ll>
#define pli pair<ll,int>
#define pii pair<int,int>
#define cd complex<double>
#define ull unsigned long long
#define base 1000000000000000000
#define fio ios::sync_with_stdio(false);cin.tie(0)

using namespace std;

const double g=10.0,eps=1e-;
const int N=+,maxn=+,inf=0x3f3f3f3f,INF=0x3f3f3f3f3f3f3f3f;

struct edge{
    int to,Next;
}e[maxn];
int cnt,head[N];
int son[N],fa[N],top[N],sz[N],id[N];
int res,w[N],re[N],dep[N];
void add(int u,int v)
{
    e[cnt].to=v;
    e[cnt].Next=head[u];
    head[u]=cnt++;
}
void init()
{
    cnt=;
    memset(head,-,sizeof head);
    memset(son,-,sizeof son);
}
void dfs1(int u,int f,int de)
{
    fa[u]=f;sz[u]=;
    dep[u]=de;
    for(int i=head[u];~i;i=e[i].Next)
    {
        int v=e[i].to;
        if(v!=f)
        {
            dfs1(v,u,de+);
            sz[u]+=sz[v];
            if(son[u]==-||sz[v]>sz[son[u]])son[u]=v;
        }
    }
}
void dfs2(int u,int f,int tp)
{
    top[u]=tp;
    id[u]=++res;
    if(son[u]!=-)dfs2(son[u],u,tp);
    for(int i=head[u];~i;i=e[i].Next)
    {
        int v=e[i].to;
        if(v!=f&&v!=son[u])dfs2(v,u,v);
    }
}
struct node{
    int cl,cr,co;
}sum[N<<];
node gao(node a,node b)
{
    if(a.cl==-)a=b;
    else if(b.cl==-);
    else
    {
        a.co+=b.co;
        if(a.cr==b.cl)a.co--;
        a.cr=b.cr;
    }
    return a;
}
int lazy[N<<];
void pushdown(int rt)
{
    if(lazy[rt]!=-)
    {
        sum[rt<<].co=sum[rt<<|].co=;
        sum[rt<<].cl=sum[rt<<|].cl=lazy[rt];
        sum[rt<<].cr=sum[rt<<|].cr=lazy[rt];
        lazy[rt<<]=lazy[rt<<|]=lazy[rt];
        lazy[rt]=-;
    }
}
void pushup(int rt)
{
    sum[rt]=gao(sum[rt<<],sum[rt<<|]);
}
void build(int l,int r,int rt)
{
    lazy[rt]=-;
    if(l==r){sum[rt].cl=sum[rt].cr=w[re[l]];sum[rt].co=;return ;}
    int m=(l+r)>>;
    build(ls);build(rs);
    pushup(rt);
}
void update(int L,int R,int c,int l,int r,int rt)
{
    if(L<=l&&r<=R)
    {
        sum[rt].co=,sum[rt].cl=sum[rt].cr=c;
        lazy[rt]=c;
        return ;
    }
    pushdown(rt);
    int m=(l+r)>>;
    if(L<=m)update(L,R,c,ls);
    if(m<R)update(L,R,c,rs);
    pushup(rt);
}
node query1(int L,int R,int l,int r,int rt)
{
    if(L<=l&&r<=R)return sum[rt];
    pushdown(rt);
    int m=(l+r)>>;
    node ans{-,-,};
    if(L<=m)ans=gao(ans,query1(L,R,ls));
    if(m<R)ans=gao(ans,query1(L,R,rs));
    return ans;
}
int query(int a,int b)
{
    int f1=top[a],f2=top[b],ans=;
    node aa{-,-,},bb{-,-,};
    while(f1!=f2)
    {
        if(dep[f1]>dep[f2])
        {
            node te=query1(id[f1],id[a],,res,);
            swap(te.cl,te.cr);
            aa=gao(aa,te);
            a=fa[f1];f1=top[a];
        }
        else
        {
            node te=query1(id[f2],id[b],,res,);
            swap(te.cl,te.cr);
            bb=gao(bb,te);
            b=fa[f2];f2=top[b];
        }
    }
    if(dep[a]<dep[b])
    {
        node te=query1(id[a],id[b],,res,);
        aa=gao(aa,te);
        swap(bb.cl,bb.cr);
        aa=gao(aa,bb);
    }
    else
    {
        node te=query1(id[b],id[a],,res,);
        swap(te.cl,te.cr);
        aa=gao(aa,te);
        swap(bb.cl,bb.cr);
        aa=gao(aa,bb);
    }
    return aa.co;
}
void change(int a,int b,int c)
{
    int f1=top[a],f2=top[b];
    while(f1!=f2)
    {
        if(dep[f1]<dep[f2])swap(f1,f2),swap(a,b);
        update(id[f1],id[a],c,,res,);
        a=fa[f1],f1=top[a];
    }
    if(dep[a]>dep[b])swap(a,b);
//    printf("%d %d %d %d\n",id[a],id[b],a,b);
    update(id[a],id[b],c,,res,);
}
int main()
{
    int n,m;
    scanf("%d%d",&n,&m);
    for(int i=;i<=n;i++)scanf("%d",&w[i]);
    init();
    for(int i=;i<n;i++)
    {
        int a,b;scanf("%d%d",&a,&b);
        add(a,b);add(b,a);
    }
    dfs1(,-,);
    dfs2(,-,);
    for(int i=;i<=res;i++)re[id[i]]=i;
    build(,res,);
    while(m--)
    {
        char op[];
        scanf("%s",op);
        if(op[]=='C')
        {
            int a,b,c;scanf("%d%d%d",&a,&b,&c);
            change(a,b,c);
        }
        else
        {
            int a,b;scanf("%d%d",&a,&b);
            printf("%d\n",query(a,b));
        }
    }
    return ;
}
/***********************
8 100
3 2 2 2  4 1 1 3
1 3
1 5
1 7
3 6
3 4
5 2
2 8
***********************/