2243: [SDOI2011]染色

题目链接

题目大意：树上区间染色，查询区间内的颜色段数

题解：记录左右端点颜色lc,rc，s表示区间答案
然后比较麻烦的就是合并区间时可能有相同颜色相连……
具体见代码吧

我的收获：用lca的姿势

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;

#define M 100005

int n,m,t,a[M],head[M],tim;
int tree[M],sz[M],son[M],dfn[M],fa[M],top[M],dep[M];

struct edge{int to,nex;}e[M*4];

void add(int u,int v){e[t]=(edge){v,head[u]};head[u]=t++;}

namespace stree{

#define ls x<<1
#define rs x<<1|1
#define lson l,m,x<<1
#define rson m+1,r,x<<1|1
#define root 1,n,1 

int rep[M<<2],s[M<<2],lc[M<<2],rc[M<<2]; 

inline void pushup(int x){
    bool tmp=!(rc[ls]^lc[rs]);
    s[x]=s[ls]+s[rs]-tmp;
    lc[x]=lc[ls],rc[x]=rc[rs];
}

void pushdown(int x,int m)//m为区间元素个数
{
    if(rep[x]==-1||m==1) return ;
    s[ls]=s[rs]=1;
    rep[ls]=rep[rs]=rep[x];
    lc[ls]=rc[ls]=rep[x];
    lc[rs]=rc[rs]=rep[x];
    rep[x]=-1;
}

void build(int l,int r,int x)
{
    rep[x]=-1,s[x]=1;//可以置为0，所以初始值设为-1 
    if(l==r){lc[x]=rc[x]=a[tree[l]];return ;}
    int m=(l+r)>>1;
    build(lson);build(rson);
    pushup(x);
}

void updata(int L,int R,int v,int l,int r,int x)
{
    if(L<=l&&r<=R){lc[x]=rc[x]=rep[x]=v;s[x]=1;return ;}//！！！ 
    pushdown(x,r-l+1);
    int m=(l+r)>>1;
    if(L<=m) updata(L,R,v,lson);
    if(R>m) updata(L,R,v,rson);
    pushup(x);
}

int query(int L,int R,int l,int r,int x) 
{
    if(L<=l&&r<=R) return s[x];
    pushdown(x,r-l+1);
    int m=(l+r)>>1,ans=0;
    bool tmp=!(rc[ls]^lc[rs])&&L<=m&&R>m;//注意要同时跨越左右子树才可以 
    if(L<=m) ans+=query(L,R,lson);
    if(R>m) ans+=query(L,R,rson);
    return ans-tmp;
}

int getc(int k,int l,int r,int x)
{
    pushdown(x,r-l+1);
    if(l==r) return lc[x];
    int m=(l+r)>>1;
    if(k<=m) return getc(k,lson);
    else return getc(k,rson);
}

}

namespace hld{

void dfs(int x)
{
    sz[x]=1;
    for(int i=head[x];i!=-1;i=e[i].nex){
        int v=e[i].to;
        if(v!=fa[x]){
            fa[v]=x;dep[v]=dep[x]+1;
            dfs(v);sz[x]+=sz[v];
            if(!son[x]||sz[v]>sz[son[x]]) son[x]=v;
        }
    }
}

void dfs2(int x,int tp){
    top[x]=tp;dfn[x]=++tim;
    tree[dfn[x]]=x;
    if(!son[x]) return ;
    dfs2(son[x],tp);
    for(int i=head[x];i!=-1;i=e[i].nex){
        int v=e[i].to;
        if(v!=fa[x]&&v!=son[x]) dfs2(v,v);
    }
}

int lca(int x,int y){
    int f1=top[x],f2=top[y];
    while(f1!=f2){
        if(dep[f1]<dep[f2]) swap(x,y),swap(f1,f2);
        x=fa[f1];f1=top[x];
    }
    return dep[x]<dep[y]?x:y;
}

}

void change(int x,int y,int z){
    int f1=top[x],f2=top[y];
    while(f1!=f2){
        stree::updata(dfn[f1],dfn[x],z,root);
        x=fa[f1];f1=top[x];
    }
    stree::updata(dfn[y],dfn[x],z,root);
}

int ask(int x,int y){
    int f1=top[x],f2=top[y],ret=0;
    while(f1!=f2){
        ret+=stree::query(dfn[f1],dfn[x],root);
        if(stree::getc(dfn[f1],root)==stree::getc(dfn[fa[f1]],root)) ret--;//判断相接的部分是否相同，跨越轻边的时候统计一下
        x=fa[f1],f1=top[x];
    }
    ret+=stree::query(dfn[y],dfn[x],root);
    return ret;
}

void work()
{
    char opt[5];
    int x,y,z,gra,ans=0;
    for(int i=1;i<=m;i++)
    {
        scanf("%s%d%d",opt,&x,&y);
        if(opt[0]=='Q') gra=hld::lca(x,y),ans+=ask(x,gra),ans+=ask(y,gra),printf("%d\n",ans-1),ans=0;
        if(opt[0]=='C') scanf("%d%d",&z),gra=hld::lca(x,y),change(x,gra,z),change(y,gra,z);
    }
}

void init()
{
    int x,y;
    cin>>n>>m;t=0;memset(head,-1,sizeof(head));
    for(int i=1;i<=n;i++) scanf("%d",&a[i]);
    for(int i=1;i<n;i++) scanf("%d%d",&x,&y),add(x,y),add(y,x);
    hld::dfs(1);hld::dfs2(1,0);stree::build(root);
}

int main()
{
    init();
    work();
}