当前位置: 代码迷 >> 综合 >> [bzoj2243][树链剖分]染色
  详细解决方案

[bzoj2243][树链剖分]染色

热度:85   发布时间:2023-12-19 05:56:27.0

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]之间。

题解

这个一眼就能看出来树剖吧。。
树剖的线段树的话维护一下这一段最左边的颜色,最右边的颜色,以及本身的颜色(不是纯色的话就为-1),打个lazy搞一下。。
修改我就不废话了大家都会吧。。
查询时x和y往上跳的时候分别维护两个值lasx和lasy,表示x,y当前查询到的最后一段上最上面那段的颜色。线段树问的时候维护两个值fir和tmp,表示当前查询这一段最下面的颜色和最上面的颜色。出来之后看一看fir和跳的这个las是不是相等,相等的话答案就要-1。之后las继承为tmp
如果你分不清我上面说的las的话请你看一看树剖跳轻重链的过程。那里会把lasx和lasy交换一下的
线段树查询的时候有一个小小的坑点。。上面说要维护一下这个段最左边和最右边的颜色,就是因为怕分开查询的时候中间连接的地方其实是一段,然后那里也是需要判一下的。。
实在听不下去我口胡的可以看一看代码。。我觉得可看性还是挺强的

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<cmath>
using namespace std;
struct node
{int x,y,next;
}a[211000];int len,last[111000];
void ins(int x,int y)
{len++;a[len].x=x;a[len].y=y;a[len].next=last[x];last[x]=len;
}
int fa[111000],dep[111000],tot[111000],son[111000],n,m;
void pre_tree_node(int x)
{son[x]=0;tot[x]=1;for(int k=last[x];k;k=a[k].next){int y=a[k].y;if(y!=fa[x]){fa[y]=x;dep[y]=dep[x]+1;pre_tree_node(y);if(tot[son[x]]<tot[y])son[x]=y;tot[x]+=tot[y];}}
}
int ys[111000],top[111000],z;
void pre_tree_edge(int x,int tp)
{ys[x]=++z;top[x]=tp;if(son[x]!=0)pre_tree_edge(son[x],tp);for(int k=last[x];k;k=a[k].next)if(son[x]!=a[k].y && fa[x]!=a[k].y)pre_tree_edge(a[k].y,a[k].y);
}
struct trnode
{int lc,rc,l,r,c;int lef,rig;bool lazy;
}tr[410000];int trlen;
void bt(int l,int r)
{int now=++trlen;tr[now].l=l;tr[now].r=r;tr[now].lc=tr[now].rc=-1;tr[now].c=tr[now].lef=tr[now].rig=0;tr[now].lazy=false;if(l<r){int mid=(l+r)/2;tr[now].lc=trlen+1;bt(l,mid);tr[now].rc=trlen+1;bt(mid+1,r);}
}
void upd(int x)
{int lc=tr[x].lc,rc=tr[x].rc;tr[lc].lazy=tr[rc].lazy=true;tr[lc].c=tr[rc].c=tr[x].c;tr[lc].lef=tr[lc].rig=tr[rc].lef=tr[rc].rig=tr[x].c;tr[x].lazy=false;
}
int lasx,lasy,fir,tmp,col[111000];
void change(int now,int l,int r,int c)
{if(tr[now].l==l && tr[now].r==r){tr[now].c=tr[now].lef=tr[now].rig=c;tr[now].lazy=true;return ;}int lc=tr[now].lc,rc=tr[now].rc;int mid=(tr[now].l+tr[now].r)/2;if(tr[now].lazy)upd(now);if(r<=mid)change(lc,l,r,c);else if(mid+1<=l)change(rc,l,r,c);else change(lc,l,mid,c),change(rc,mid+1,r,c);if(tr[lc].c==tr[rc].c)tr[now].c=tr[lc].c;else tr[now].c=-1;tr[now].lef=tr[lc].lef;tr[now].rig=tr[rc].rig;
}
int findsum(int now,int l,int r,int lef,int rig)
{if(tr[now].c!=-1){if(r>=rig)fir=tr[now].c;//问到了最下面的那个 if(l<=lef)tmp=tr[now].c;//最上面的那个 return 1;}int lc=tr[now].lc,rc=tr[now].rc;int mid=(tr[now].l+tr[now].r)/2;if(tr[now].lazy)upd(now);if(r<=mid)return findsum(lc,l,r,lef,rig);else if(mid+1<=l)return findsum(rc,l,r,lef,rig);else {int retx=findsum(lc,l,mid,lef,rig),rety=findsum(rc,mid+1,r,lef,rig);if(tr[lc].rig==tr[rc].lef)return retx+rety-1;//中间其实是一段的情况 return retx+rety;}
}
void chmul(int x,int y,int c)
{int tx=top[x],ty=top[y];while(tx!=ty){if(dep[ty]<dep[tx])swap(x,y),swap(tx,ty);change(1,ys[ty],ys[y],c);y=fa[ty];ty=top[y];}if(x==y){change(1,ys[x],ys[x],c);return ;}else{if(dep[x]>dep[y])swap(x,y);change(1,ys[x],ys[y],c);return ;    }
}
int sol(int x,int y)
{int tx=top[x],ty=top[y],ans=0;lasx=lasy=-999999999;while(tx!=ty){if(dep[ty]<dep[tx])swap(x,y),swap(tx,ty),swap(lasx,lasy);ans+=findsum(1,ys[ty],ys[y],ys[ty],ys[y]);if(fir==lasy)ans--;lasy=tmp;y=fa[ty];ty=top[y];}if(x==y){ans+=findsum(1,ys[x],ys[x],ys[x],ys[x]);if(fir==lasx)ans--;if(fir==lasy)ans--;return ans;}else{if(dep[x]>dep[y])swap(x,y),swap(lasx,lasy);ans+=findsum(1,ys[x],ys[y],ys[x],ys[y]);//x在上 y在下 if(fir==lasy)ans--;if(tmp==lasx)ans--;return ans;}
}
int main()
{scanf("%d%d",&n,&m);for(int i=1;i<=n;i++)scanf("%d",&col[i]);len=0;memset(last,0,sizeof(last));for(int i=1;i<n;i++){int x,y;scanf("%d%d",&x,&y);ins(x,y);ins(y,x);}fa[1]=0;dep[1]=0;pre_tree_node(1);z=0;pre_tree_edge(1,1);trlen=0;bt(1,z);for(int i=1;i<=n;i++)change(1,ys[i],ys[i],col[i]);while(m--){char ss[10];int x,y,c;scanf("%s%d%d",ss+1,&x,&y);if(ss[1]=='C'){scanf("%d",&c);chmul(x,y,c);}else printf("%d\n",sol(x,y));}return 0;
}