[BZOJ2243]染色 做题笔记

·· / ·– ·· ·-·· ·-·· / ·–· · ·-· ··· ·· ··· - / ··- -· - ·· ·-·· / ·· / ·– ·· -·

题目来源：http://www.lydsy.com/JudgeOnline/problem.php?id=2243

线段树维护色块数量，区间最左端的颜色，区间最右端的颜色。合并区间时，如果左区间的最右端的颜色等于右区间的最左端的颜色，注意-1。

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int N=100010,inf=0x7fffffff;
#define lch now<<1
#define rch now<<1|1
struct Tree {
int col,l,r,num,lc,rc;
}I[N<<2];
int n,m,num,a,b,tot,L,R,val,pos,cnt;
int xx,yy,head
,ver[N<<1],nxt[N<<1];
int son
,top
,w
,C
;
int dep
,fa
,sz
,id
;
int Lc,Rc,u,v,c;
void add(int u,int v) {
ver[++tot]=v;nxt[tot]=head[u];head[u]=tot;
}
void dfs_1 (int u,int from) {
sz[u]=1,son[u]=0,fa[u]=from;
for (int i=head[u];i;i=nxt[i]) {
int v=ver[i];
if (v==fa[u]) continue;
dep[v]=dep[u]+1; fa[v]=u;
dfs_1(v,u); sz[u]+=sz[v];
if (!son[u]||sz[son[u]]<sz[v]) son[u]=v;
}
}
void dfs_2 (int u,int st) {
w[u]=++cnt;top[u]=st;id[cnt]=u;
if (son[u]) dfs_2(son[u],st);
for (int i=head[u];i;i=nxt[i]) {
int v=ver[i];
if (v==son[u]||v==fa[u]) continue;
dfs_2(v,v);
}
}
void pushup (int now) {
I[now].lc=I[lch].lc;I[now].rc=I[rch].rc;
int ans=I[lch].num+I[rch].num;
if (I[lch].rc==I[rch].lc) ans--;
I[now].num=ans;
}
void pushdown (int now) {
if (!I[now].col) return;//
I[lch].col=I[rch].col=I[now].col;
I[lch].num=I[rch].num=1;
I[lch].lc=I[lch].rc=I[now].lc;//
I[rch].lc=I[rch].rc=I[now].lc;//
I[now].col=0;
}
void build (int now,int l,int r) {
I[now].l=l;I[now].r=r;I[now].num=0;
if (l==r) return ;
int mid=(l+r)>>1;
build(lch,l,mid);build(rch,mid+1,r);
}
void change (int now,int l,int r,int x) {
if (I[now].l==l&&I[now].r==r) {
I[now].num=I[now].col=1;
I[now].lc=I[now].rc=x;
return ;
}
pushdown(now);
int mid=(I[now].l+I[now].r)>>1;
if (r<=mid) change(lch,l,r,x);
else if (l>mid) change(rch,l,r,x);
else change(lch,l,mid,x),change(rch,mid+1,r,x);
pushup(now);
}
int query (int now,int l,int r,int L,int R) {
if (I[now].l==L) Lc=I[now].lc;
if (I[now].r==R) Rc=I[now].rc;
if (I[now].l==l&&I[now].r==r) return I[now].num;
pushdown(now);
int mid=(I[now].l+I[now].r)>>1;
if (r<=mid) return query(lch,l,r,L,R);
else if (l>mid) return query(rch,l,r,L,R);
else {
int ans=query(lch,l,mid,L,R)+query(rch,mid+1,r,L,R);
if (I[lch].rc==I[rch].lc) ans--;
return ans;
}
pushup(now);
}
int Get (int u,int v,int opt,int c) {
int ans=0;
if (opt==1) {
while (top[u]!=top[v]) {
if (dep[top[u]]<dep[top[v]]) swap(u,v);
change(1,w[top[u]],w[u],c);
u=fa[top[u]];
}
if (dep[u]>dep[v]) swap(u,v);
change(1,w[u],w[v],c);
}
else {
int ans1=-1,ans2=-1;
while (top[u]!=top[v]) {
if (dep[top[u]]<dep[top[v]]) swap(u,v),swap(ans1,ans2);
int L=w[top[u]],R=w[u];
ans+=query(1,L,R,L,R);
if (Rc==ans1) ans--;
u=fa[top[u]],ans1=Lc;
}
if (dep[u]<dep[v]) swap(u,v),swap(ans1,ans2);
ans+=query(1,w[v],w[u],w[v],w[u]);
if (Rc==ans1) ans--;//注意判断
if (Lc==ans2) ans--;//
}
return ans;
}
int main () {
char str[30];
scanf("%d%d",&n,&m);
cnt=0;tot=1;
for (int i=1;i<=n;i++) scanf("%d",&C[i]);
for (int i=1;i<n;i++) {
scanf("%d%d",&u,&v);
add(u,v); add(v,u);
}
dfs_1(1,0);
dfs_2(1,1);
build(1,1,n);
for (int i=1;i<=n;i++)
change(1,w[i],w[i],C[i]);
while (m--) {
scanf("%s",str);
if (str[0]=='C') {
scanf("%d%d%d",&u,&v,&c);
Get(u,v,1,c);
}
else {
scanf("%d%d",&u,&v);
printf("%d\n",Get(u,v,2,0));
}
}
return 0;
}