染色 SDOI2011
描述:
给定一棵有n个节点的无根树和m个操作,操作有2类:
1、将节点a到节点b路径上所有点都染成颜色c;
2、询问节点a到节点b路径上的颜色段数量(连续相同颜色被认为是同一段),如“112221”由3段组成:“11”、“222”和“1”。
请你写一个程序依次完成这m个操作。
输入:
第一行包含2个整数n和m,分别表示节点数和操作数;
第二行包含n个正整数表示n个节点的初始颜色
下面n-1行每行包含两个整数x和y,表示x和y之间有一条无向边。
下面m行每行描述一个操作:
“C a b c”表示这是一个染色操作,把节点a到节点b路径上所有点(包括a和b)都染成颜色c;
“Q a b”表示这是一个询问操作,询问节点a到节点b(包括a和b)路径上的颜色段数量。
输出:
对于每个询问操作,输出一行答案。
样例输入
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
样例输出
3
1
2
显然,数据规模是骗人的,1<=n<=1e5才是正解哦
解题思路:
树链剖分+线段树维护。
树链部分除了查询全模板,这里不过多详述,说一下查询,查询时往上跳(意会就好)的时候需要保存下lastx和lasty。配合查询x和y查询的结果更新答案。考虑到一一对应性,交换时两组值都要交换。
线段树部分需要维护区间最左边的颜色,最右边的颜色,颜色段的个数以及一个万恶的lazy标记(代码中用cg表示)。大区间的颜色段数=左区间颜色段数+右区间颜色段数,如果左区间最右端等于右区间最左端,大区间颜色段数减一。
本题还有一大坑点,注意建树时不能把颜色更新,只更新区间标记和cg即可,同时别忘了sum归零。建树完成后再更新每个节点的值才不会把树上不同链的颜色更新在一起。
接下来也就是最最恶心的代码实现和排错了,虽然花费我半天时间,但时间也不是白费,至少我对树链剖分的运行机制和坑点有了更好的理解,嗯,下次会更好,加油!
代码附上
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
#define INF 0x3f3f3f3f
#define N 100010
#define LD (o<<1)
#define RD ((o<<1)|1)
struct Tree{
int l,r,sum;
int lc,rc;//记录最左最右的颜色
int cg;//记录更改
}T[4*N];//线段树
struct Edge{int v,next;}E[N];//边
struct Node{
int dep,siz,fa,w,hson,num,top;
Node(){dep=siz=fa=w=hson=num=top=0;}
}P[N];//点
/*
dep深度
siz节点数
fa父亲
w节点权值
hson重孩子
num在线段树编号
top表示每条链第一个位置
*/
int first[N],pred[N];
/*
first 邻接表头指针
pred线段树的编号是原来什么编号
*/
int n,m,last,tot;//tot记录线段树总编号
void init(){
last=tot=0;
memset(first,0,sizeof(first));
memset(pred,0,sizeof(pred));
}
inline int read(){
long long data=0,w=1; char ch=0;
while(ch!='-' && (ch<'0' || ch>'9')) ch=getchar();
if(ch=='-') w=-1,ch=getchar();
while(ch>='0' && ch<='9'){data=data*10+ch-'0';ch=getchar();}
return data*w;
}
void add(int u,int v){
E[++last].v=v;
E[last].next=first[u];
first[u]=last;
}
void dfs1(int s){
P[s].siz=1;
P[s].hson=0;
for(int i=first[s];i;i=E[i].next){
int &son=E[i].v;
if(son==P[s].fa)continue;
P[son].dep=P[s].dep+1;
P[son].fa=s;
dfs1(son);
P[s].siz+=P[son].siz;
if(P[P[s].hson].siz<P[son].siz)
P[s].hson=son;
}
}
void dfs2(int s,int tp){
P[s].top=tp;
P[s].num=++tot;
pred[tot]=s;
if(P[s].hson)dfs2(P[s].hson,tp);
for(int i=first[s];i;i=E[i].next){
int &son=E[i].v;
if(son!=P[s].fa&&son!=P[s].hson)
dfs2(son,son);
}
}
void pushup(int o){
T[o].lc=T[LD].lc;T[o].rc=T[RD].rc;
T[o].sum=T[LD].sum+T[RD].sum;
if(T[LD].rc==T[RD].lc)T[o].sum--;//左右区间相邻处颜色相同合为同一色段
}
void pushdown(int o){
if(T[o].cg){
int t=T[o].cg;
T[LD].cg=T[RD].cg=t;
T[LD].lc=T[LD].rc=t;
T[RD].lc=T[RD].rc=t;
T[o].lc=T[o].rc=t;
T[LD].sum=T[RD].sum=T[o].sum=1;
T[o].cg=0;
}
}
void build(int o,int l,int r){//只建树不加权值避免加错线段
T[o].l=l;T[o].r=r;
if(l==r){
T[o].lc=T[o].rc=-1;//节点颜色更新为-1
T[o].sum=0;T[o].cg=-1;
return;
}
int mid=(l+r)>>1;
build(LD,l,mid);
build(RD,mid+1,r);
}
void update(int o,int l,int r,int val){
if(T[o].l>r||T[o].r<l)return;
if(l<=T[o].l&&T[o].r<=r){
T[o].cg=T[o].lc=T[o].rc=val;
T[o].sum=1;
return;
}
pushdown(o);
int mid=(T[o].l+T[o].r)>>1;
if(l<=mid)update(LD,l,r,val);
if(r>=mid+1)update(RD,l,r,val);
pushup(o);
}
Tree query(int o,int l,int r){
if(l<=T[o].l&&T[o].r<=r)return T[o];
pushdown(o);
int mid=(T[o].l+T[o].r)>>1;
if(mid>=r)return query(LD,l,r);
if(mid<l)return query(RD,l,r);
Tree ans;
Tree LL=query(LD,l,mid);
Tree RR=query(RD,mid+1,r);
ans.l=LL.l;ans.r=RR.r;
ans.lc=LL.lc;ans.rc=RR.rc;
ans.sum=LL.sum+RR.sum;
if(LL.rc==RR.lc)ans.sum--;//合并区间注意边界值影响
return ans;
}
int get_sum(int x,int y){
int lastx=-1,lasty=-1,ans=0;//保存两个标记量,更新ans时使用
while(P[x].top!=P[y].top){//不在一条重链时
if(P[x].num<P[y].num)swap(x,y),swap(lastx,lasty);
Tree tmp=query(1,P[P[x].top].num,P[x].num);
ans+=tmp.sum;
if(lastx==tmp.rc)ans--;//左右区间相邻处颜色相同合为同一色段
lastx=tmp.lc;//更新标记
//注意此时只用更新lastx,因为lasty在与lastx交换的过程中已经进行了同步更新
x=P[P[x].top].fa;//往上跳
}
if(P[x].num>P[y].num)swap(x,y),swap(lastx,lasty);
Tree tmp=query(1,P[x].num,P[y].num);
ans+=tmp.sum;
if(lastx==tmp.lc)ans--;//因为在一条重链上,需要判断两遍
if(lasty==tmp.rc)ans--;
return ans;
}
void get_update(int x,int y,int val){
while(P[x].top!=P[y].top){
if(P[x].num<P[y].num)swap(x,y);
update(1,P[P[x].top].num,P[x].num,val);
x=P[P[x].top].fa;
}
if(P[x].num>P[y].num)swap(x,y);
update(1,P[x].num,P[y].num,val);
}
int main(){
n=read();m=read();
for(int i=1;i<=n;++i)P[i].w=read();
for(int i=1;i<n;++i){
int a=read(),b=read();
add(a,b);
add(b,a);
}
dfs1(1);
dfs2(1,1);
build(1,1,n);
for(int i=1;i<=n;++i)update(1,P[i].num,P[i].num,P[i].w);
//建树后更新每一个点的颜色
for(int i=1;i<=m;++i){
char t[3];int a,b,c;
scanf("%s",t);a=read();b=read();
if(t[0]=='Q')printf("%d\n",get_sum(a,b));
if(t[0]=='C'){
c=read();
get_update(a,b,c);
}
}
return 0;
}