【SDOI2011】染色

Description

给出一棵树，每个点有颜色。
m次操作，每次操作把一条路径上的颜色都修改成c，
或询问一条路径上的颜色段数。
n,m<=10^5

Solution

简单的链剖题。
每个区间维护颜色段数和左右端点的颜色就好了。
然而这道题会爆栈！！！而且要写有方向的往上跳~~
于是我方了~~
然后就滚回去写LCT了。
为了练手我就写了个静态LCT（然并卵）
然后发现x=y时跪了。。。
有点奥妙重重啊233

Code

#include<cstdio>
#include<cstring>
#include<algorithm>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define rep(i,a) for(int i=last[a];i;i=next[i])
#define N 100005
using namespace std;
int t[N][2],f[N],key[N],le[N],ri[N],sz[N],lazy[N],d[N],p[N],lca;
int to[N*2],next[N*2],last[N],fa[N];
int n,m,x,y,z,l,ans,deep[N],q[N];
char s[1];
void add(int x,int y) {
    to[++l]=y;next[l]=last[x];last[x]=l;
}
int son(int x) {
    return t[f[x]][1]==x;
}
void updata(int x) {
    if (t[x][0]) le[x]=le[t[x][0]];
    else le[x]=key[x];
    if (t[x][1]) ri[x]=ri[t[x][1]];
    else ri[x]=key[x];
    sz[x]=sz[t[x][0]]+sz[t[x][1]]+1;
    if (t[x][0]&&ri[t[x][0]]==key[x]) sz[x]--;
    if (t[x][1]&&le[t[x][1]]==key[x]) sz[x]--;
}
void back(int x,int y) {
    if (!x) return;
    sz[x]=1;le[x]=ri[x]=lazy[x]=key[x]=y;
}
void down(int x) {
    if (lazy[x]) {
        back(t[x][0],lazy[x]);
        back(t[x][1],lazy[x]);
        lazy[x]=0;
    }
}
void remove(int x,int y) {
    do {
        d[++d[0]]=x;x=f[x];
    } while (x!=y);
    while (d[0]) down(d[d[0]--]);
} 
void rotate(int x) {
    int y=f[x],z=son(x);f[x]=f[y];
    if (f[x]) t[f[x]][son(y)]=x;
    else p[x]=p[y],p[y]=0;
    if (t[x][1-z]) f[t[x][1-z]]=y;
    f[y]=x;t[y][z]=t[x][1-z];t[x][1-z]=y;
    updata(y);updata(x);
}
void splay(int x,int y) {
    remove(x,y);
    while (f[x]!=y) {
        if (f[f[x]]!=y) 
            if (son(x)==son(f[x])) rotate(f[x]);
            else rotate(x);
        rotate(x);
    }
}
void access(int x) {
    int y=0;
    while (x) {
        splay(x,0);
        f[t[x][1]]=0;p[t[x][1]]=x;
        t[x][1]=y;f[y]=x;p[y]=0;
        updata(x);y=x;x=p[x];
    }lca=y;
}
int main() {
    scanf("%d%d",&n,&m);
    fo(i,1,n) scanf("%d",&key[i]);
    fo(i,1,n-1) scanf("%d%d",&x,&y),add(x,y),add(y,x);
    int i=0,j=1;deep[1]=q[1]=1;
    while (i<j) {
        rep(k,q[++i]) if (to[k]!=fa[q[i]]) {
            fa[to[k]]=q[i];deep[to[k]]=deep[q[i]]+1;
            p[to[k]]=q[i];q[++j]=to[k];
        }
    }
    for(;m;m--) {
        scanf("%s%d%d",s,&x,&y);
        if(s[0]=='C'){
            scanf("%d",&z);
            if (x==y) {splay(x,0);key[x]=z;continue;}
            if(deep[x]<deep[y])swap(x,y);
            access(x);access(y);splay(x,0);
            key[lca]=z;back(x,z);
            back(t[lca][1],z);updata(lca);
        }
        else{
            if (x==y) {printf("1\n");continue;}
            if(deep[x]<deep[y])swap(x,y);
            access(x);access(y);splay(x,0);
            ans=sz[x]+sz[t[lca][1]]+1;
            if(le[x]==key[lca])ans--;
            if(key[lca]==le[t[lca][1]])ans--;
            printf("%d\n",ans);
        }
    }
}