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;}