【Bzoj2243】染色

题意

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

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

2、询问节点a到节点b路径上的颜色段数量（连续相同颜色被认为是同一段），如“112221”由3段组成：“11”、“222”和“1”。

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

---

应该是我太水了来着…？打了一小时调了一天，主要就是注意合并时判断中间部分的端点颜色，如果一样的话统计时就要减1。然后在get函数中也一样，接到上一段的时候要判断端点的颜色，这题还是很有价值的…然后那个数组ni，就是逆的意思，表示线段树中的结点的原节点…

#include <cstdio>
#include <algorithm>

#define Rep(i,s,t) for(int i=s;i<=t;i++)
#define For(i,s,t) for(int i=s;i;i=t)

using namespace std;

const int maxx = 100000 + 25;

int head[maxx],to[maxx<<1],nxt[maxx<<1],a[maxx],ni[maxx];
int top[maxx],dpt[maxx],son[maxx],ftr[maxx],rnk[maxx],size[maxx];
int T[maxx<<2],col[maxx<<2],Add[maxx<<2],lc[maxx<<2],rc[maxx<<2];
int n,m,num,cnt,x,y,z;
char f[5];

namespace Y{

    void Ins(int x,int y){
        to[++num] = y;nxt[num] = head[x];head[x] = num;
        to[++num] = x;nxt[num] = head[y];head[y] = num;
    }

    void Dfs1(int x){
        size[x] = 1;
        For(i , head[x] , nxt[i]){
            int now = to[i];if(now == ftr[x]) continue;
            ftr[now] = x;dpt[now] = dpt[x] + 1;
            Dfs1(now);size[x] += size[now];
            if(size[now] > size[son[x]]) son[x] = now;
        }
    }

    void Dfs2(int x,int brn){
        rnk[x] = ++cnt;top[x] = brn;
        ni[cnt] = x;
        if(son[x]) Dfs2(son[x],brn);
        For(i , head[x] , nxt[i])
            if(to[i]!=ftr[x] && to[i]!=son[x])
                Dfs2(to[i],to[i]);
    }

    void upt(int i){
        lc[i] = lc[i<<1];rc[i] = rc[i<<1|1];
        T[i] = T[i<<1] + T[i<<1|1]; 
        if(rc[i<<1] == lc[i<<1|1]) T[i]--;
    }

    void pushdown(int i){
        Add[i<<1] = Add[i<<1|1] = Add[i];
        lc[i<<1] = rc[i<<1] = lc[i<<1|1] = rc[i<<1|1] = Add[i];
        T[i<<1] = T[i<<1|1] = 1;Add[i] = 0;
    }

    void Build(int i,int l,int r){
        if(l == r) {lc[i] = rc[i] = a[ni[l]];T[i] = 1;return;}
        int mid = (l+r) >> 1;
        Build(i<<1,l,mid);Build(i<<1|1,mid+1,r);
        upt(i);
    }

    void modify(int i,int x,int y,int l,int r,int k){
        if(x <= l && r <= y){T[i]=1;lc[i]=rc[i]=k;Add[i]=k;return;}
        if(Add[i]) pushdown(i);int mid = (l+r) >> 1;
        if(x <= mid) modify(i<<1,x,y,l,mid,k);
        if(y >  mid) modify(i<<1|1,x,y,mid+1,r,k);
        upt(i);
    }

    int Query(int i,int x,int y,int l,int r){
        if(x <= l && r <= y) return T[i];int ans = 0;
        if(Add[i]) pushdown(i);int mid = (l+r) >> 1;
        if(x <= mid) ans += Query(i<<1,x,y,l,mid);
        if(y >  mid) ans += Query(i<<1|1,x,y,mid+1,r);
        if(x <= mid && y > mid && rc[i<<1] == lc[i<<1|1]) return ans-1; 
        return ans;
    }

    void update(int x,int y,int k){
        while(top[x] != top[y]){
            if(dpt[top[x]] > dpt[top[y]]) x^=y^=x^=y;
            modify(1,rnk[top[y]],rnk[y],1,n,k);
            y = ftr[top[y]];
        }
        if(rnk[x] > rnk[y]) x^=y^=x^=y;
        modify(1,rnk[x],rnk[y],1,n,k);
    }

    int col(int i,int pos,int l,int r){
        if(l == r) return lc[i];
        if(Add[i]) pushdown(i);
        int mid = (l+r) >>1;
        if(pos <= mid) return col(i<<1,pos,l,mid);
        else return col(i<<1|1,pos,mid+1,r);
    }

    int get(int x,int y){
        int ans = 0;
        while(top[x] != top[y]){
            if(dpt[top[x]] > dpt[top[y]]) x^=y^=x^=y;
            int a = col(1,rnk[top[y]],1,n),b = col(1,rnk[ftr[top[y]]],1,n);
            ans += Query(1,rnk[top[y]],rnk[y],1,n);
            if(a == b) ans -- ; 
            y = ftr[top[y]];
        }
        if(rnk[x] > rnk[y]) x^=y^=x^=y;
        ans += Query(1,rnk[x],rnk[y],1,n);
        return ans;
    }

}

using namespace Y;

int main(){
    scanf("%d%d",&n,&m);Rep(i,1,n) scanf("%d",&a[i]);
    Rep(i,1,n-1) scanf("%d%d",&x,&y),Ins(x,y);
    Dfs1(1);Dfs2(1,1);Build(1,1,n);
    while( m-- ){
        scanf("%s",f);
        if(f[0] == 'Q') scanf("%d%d",&x,&y),printf("%d\n",get(x,y));
        if(f[0] == 'C') scanf("%d%d%d",&x,&y,&z),update(x,y,z);
    }
    return 0;
}