BZOJ2243 染色

題意：

給定一棵有n個節點的無根樹和m個操作，操作有2類：

1、將節點a到節點b路徑上所有點都染成顏色c；

2、詢問節點a到節點b路徑上的顏色段數量（連續相同顏色被認爲是同一段），如“ 112221 ” 由3段組成：“ 11 ” 、“ 222 ” 和“ 1 ” 。

請你寫一個程序依次完成這m個操作。

題解

LCT

維護最左/右端點顏色

合併即可

代碼

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<iostream>
#define L(i) (T[i].s[0])
#define R(i) (T[i].s[1])
#define F(i) (T[i].fa)
#define Rc(i) (T[i].rc)
#define Lc(i) (T[i].lc)
#define C(i) (T[i].c)
#define Loc(i) (R(F(i))==i)
#define For(i,j,k) for(int i=(j);i<=(int)k;i++)
#define Forr(i,j,k) for(int i=(j);i>=(int)k;i--)
#define Set(a,b) memset(a,b,sizeof(a))
using namespace std;
const int N=100010;
inline void read(int &x){
    x=0;char c=getchar();int f=0;
    while(c<'0'||c>'9')f|=(c=='-'),c=getchar();
    while(c>='0'&&c<='9')x=x*10+c-'0',c=getchar();
    if(f)x=-x;
}
int n,m;
struct node{
    int s[2],fa,rev,c,lc,rc,sum,mk;
};
struct LCT{
    node T[N];   
    inline void clear(){Set(T,0);}
    inline void pushup(int x){
        T[x].sum=T[L(x)].sum+T[R(x)].sum+1;
        if(L(x))Lc(x)=Lc(L(x)),T[x].sum-=(T[x].c==Rc(L(x)));else Lc(x)=C(x);
        if(R(x))Rc(x)=Rc(R(x)),T[x].sum-=(T[x].c==Lc(R(x)));else Rc(x)=C(x);
    }
    inline void Swap(int x){
        swap(L(x),R(x)),swap(Lc(x),Rc(x));
    }
    inline void mark(int x,int c){
        T[x].c=Lc(x)=Rc(x)=T[x].mk=c,T[x].sum=1;
    }
    inline void pushdown(int x){
        if(T[x].rev){
            T[x].rev=0,T[L(x)].rev^=1,T[R(x)].rev^=1;
            Swap(L(x));Swap(R(x));
        }
        if(T[x].mk){
            int mk=T[x].mk;
            if(L(x))mark(L(x),mk);
            if(R(x))mark(R(x),mk);T[x].mk=0;
        }
    }
    inline bool isrt(int x){
        return R(F(x))!=x&&L(F(x))!=x;
    }
    void Pushdown(int x){
        if(!isrt(x))Pushdown(F(x));
        pushdown(x);
    }
    inline void Rotate(int x){
        int A=F(x),B=F(A),l=Loc(x),r=l^1,d=Loc(A);
        if(!isrt(A))T[B].s[d]=x;F(x)=B;
        F(A)=x,F(T[x].s[r])=A,T[A].s[l]=T[x].s[r],T[x].s[r]=A;
        pushup(A),pushup(x);
    }
    inline void splay(int x){   
        Pushdown(x);
        while(!isrt(x)){
            if(!isrt(F(x)))Rotate(x);
            Rotate(x);
        }
    }
    inline void access(int x){
        for(int i=0;x;i=x,x=F(x))
            splay(x),R(x)=i,pushup(x);
    }
    inline void reverse(int x){
        access(x),splay(x),T[x].rev^=1;Swap(x);
    }
    inline void split(int x,int y){
        reverse(x),access(y),splay(y);
    }
    inline void modify(int x,int y,int z){
        split(x,y),mark(y,z);
    }
    inline int query(int x,int y){
        split(x,y);
        return T[y].sum; 
    }
}t;
int Begin[N],Next[N<<1],to[N<<1],e;
inline void add(int x,int y){
    to[++e]=y,Next[e]=Begin[x],Begin[x]=e;
}
void dfs(int u,int fa){
    for(int i=Begin[u];i;i=Next[i]){
        int v=to[i];
        if(v==fa)continue;
        t.T[v].fa=u;
        dfs(v,u);
    }
}
char s[10];
int main(){
    int u,v,c;
    read(n),read(m);
    For(i,1,n)read(c),t.mark(i,c+1),t.T[i].mk=0;
    For(i,1,n-1){
        read(u),read(v);
        add(u,v),add(v,u);
    }
    dfs(1,0);
    while(m--){
        scanf("%s",s),read(u),read(v);
        if(s[0]=='C')read(c),t.modify(u,v,c+1);
        else printf("%d\n",t.query(u,v));
    }
    return 0;
}