2243: [SDOI2011]染色
Time Limit: 20 Sec Memory Limit: 512 MB
Submit: 6651 Solved: 2432
[Submit][Status][Discuss]
Description
给定一棵有n个节点的无根树和m个操作，操作有2类：
1、将节点a到节点b路径上所有点都染成颜色c；
2、询问节点a到节点b路径上的颜色段数量（连续相同颜色被认为是同一段），如“112221”由3段组成：“11”、“222”和“1”。
请你写一个程序依次完成这m个操作。
Input
第一行包含2个整数n和m，分别表示节点数和操作数；
第二行包含n个正整数表示n个节点的初始颜色
下面 行每行包含两个整数x和y，表示x和y之间有一条无向边。
下面 行每行描述一个操作：
“C a b c”表示这是一个染色操作，把节点a到节点b路径上所有点（包括a和b）都染成颜色c；
“Q a b”表示这是一个询问操作，询问节点a到节点b（包括a和b）路径上的颜色段数量。
Output
对于每个询问操作，输出一行答案。
Sample Input
6 5
2 2 1 2 1 1
1 2
1 3
2 4
2 5
2 6
Q 3 5
C 2 1 1
Q 3 5
C 5 1 2
Q 3 5
Sample Output
3
1
2
HINT
数N<=10^5，操作数M<=10^5，所有的颜色C为整数且在[0, 10^9]之间。
Source
裸树链剖分
太愚蠢了我说怎么全WA最后发现建树时给线段树的节点分配颜色弄错了
线段树就是普通的颜色覆盖而已，注意两个重链之间判断颜色相同
加了那个普通的标记剪枝，就bzoj rank16了
//
// main.cpp
// sdoi2011染色
//
// Created by Candy on 2016/12/14.
// Copyright © 2016年 Candy. All rights reserved.
// #include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
#define lc o<<1
#define rc o<<1|1
#define m ((l+r)>>1)
#define lson o<<1,l,m
#define rson o<<1|1,m+1,r
const int N=1e5+;
int read(){
char c=getchar();int x=,f=;
while(c<''||c>''){if(c=='-')f=-; c=getchar();}
while(c>=''&&c<=''){x=x*+c-''; c=getchar();}
return x*f;
}
int n,Q,ww[N],w[N],x,y,a,b,c;
char s[];
struct edge{
int v,ne;
}e[N<<];
int h[N],cnt;
inline void ins(int u,int v){
cnt++;
e[cnt].v=v;e[cnt].ne=h[u];h[u]=cnt;
cnt++;
e[cnt].v=u;e[cnt].ne=h[v];h[v]=cnt;
}
int top[N],size[N],tid[N],tot,fa[N],deep[N],mx[N];
void dfs(int u){
size[u]++;
for(int i=h[u];i;i=e[i].ne){
int v=e[i].v;
if(v==fa[u]) continue;
fa[v]=u;deep[v]=deep[u]+;
dfs(v);
size[u]+=size[v];
if(size[v]>size[mx[u]]) mx[u]=v;
}
}
void dfs(int u,int anc){
if(!u) return;
tid[u]=++tot;top[u]=anc;
dfs(mx[u],anc);
for(int i=h[u];i;i=e[i].ne)
if(e[i].v!=fa[u]&&e[i].v!=mx[u]) dfs(e[i].v,e[i].v);
} struct node{
int lv,rv,cnt,tag;
}t[N<<];
inline void merge(int o){
t[o].tag=-;
t[o].lv=t[lc].lv;
t[o].rv=t[rc].rv;
t[o].cnt=t[lc].cnt+t[rc].cnt-(t[lc].rv==t[rc].lv);
}
inline void paint(int o,int v){
t[o].tag=t[o].lv=t[o].rv=v;
t[o].cnt=;
}
void build(int o,int l,int r){
if(l==r) paint(o,w[l]);
else{
build(lson);
build(rson);
merge(o);
}
}
inline void pushDown(int o){
if(t[o].tag!=-){
paint(lc,t[o].tag);
paint(rc,t[o].tag);
t[o].tag=-;
}
}
void change(int o,int l,int r,int ql,int qr,int v){
if(ql<=l&&r<=qr) paint(o,v);
else{
pushDown(o);
if(ql<=m) change(lson,ql,qr,v);
if(m<qr) change(rson,ql,qr,v);
merge(o);
}
}
int query(int o,int l,int r,int ql,int qr){
if(ql<=l&&r<=qr) return t[o].cnt;
else if(t[o].tag!=-) return ;
else{
pushDown(o);
int ans=;
if(ql<=m) ans+=query(lson,ql,qr);
if(m<qr) ans+=query(rson,ql,qr);
if(ql<=m&&m<qr&&t[lc].rv==t[rc].lv) ans--;
return ans;
}
}
int getcol(int o,int l,int r,int p){
if(l==r) return t[o].lv;
else if(t[o].tag!=-) return t[o].tag;
else{
if(p<=m) return getcol(lson,p);
else return getcol(rson,p);
}
}
void sol1(int x,int y,int c){
while(top[x]!=top[y]){
if(deep[top[x]]<deep[top[y]]) swap(x,y);
change(,,n,tid[top[x]],tid[x],c);
x=fa[top[x]];
}
if(tid[x]>tid[y]) swap(x,y);
change(,,n,tid[x],tid[y],c);
}
int sol2(int x,int y){
int ans=;
while(top[x]!=top[y]){
if(deep[top[x]]<deep[top[y]]) swap(x,y);
ans+=query(,,n,tid[top[x]],tid[x]);
if(getcol(,,n,tid[top[x]])==getcol(,,n,tid[fa[top[x]]])) ans--;
x=fa[top[x]];
}
if(tid[x]>tid[y]) swap(x,y);
ans+=query(,,n,tid[x],tid[y]);
return ans;
}
int main(int argc, const char * argv[]) {
n=read();Q=read();
for(int i=;i<=n;i++) ww[i]=read();
for(int i=;i<=n-;i++) x=read(),y=read(),ins(x,y);
dfs();dfs(,);
for(int i=;i<=n;i++) w[tid[i]]=ww[i];
build(,,n);
while(Q--){
scanf("%s",s);a=read();b=read();
if(s[]=='C'){
c=read();sol1(a,b,c);
}else printf("%d\n",sol2(a,b));
}
return ;
}