题目大意:
给一棵树,每个节点有一个颜色。写一个程序支持把两个点路径上的所有点染成一个颜色,查询两点之间的色段数量。
解题思路:
树链剖分+线段树
首先它是一颗树,而且是修改和查询两点路径上的颜色,可以想到树链剖分。
查询颜色段数可以用线段树维护区间颜色段数。
这道题涉及到区间合并,所以在线段树和lca的时候需要多记录一些东西,当前区间的最左边的颜色,最右边的颜色,已经求出的区间的左边的颜色。
我采用的做法是先求出lca,然后从左右两边各爬一边,就能有序的将所有的区间连在一起,当然大神告诉我可以左右同时爬,但是我不太会。
注意事项:
1、在树链剖分求lca的时候一定要写函数接口,不要把大段代码放在lca里,代码非常凌乱,不易差错;
2、打了标记要记得下传,时时刻刻思考要不要下传标记;
3、每次区间合并的时候要判断是否连接的部分是相同的颜色,如果相同要ans--;
4、从一侧爬的时候处理lca,另一侧就要避开lca,而且左右两条链合并的时候要判断颜色是否相同。
#include<cstdio>
#include<cstdlib>
#include<iostream>
#include<string>
#include<cstring>
#include<algorithm>
#include<cmath>
#define N 100100
using namespace std;
struct tree{
int l,r,sum,ls,rs,lz;
}d[N*8];
int n,m,x,y,z,a,b;
char s[5];
int q[N];
int fir[N],nes[N*2],v[N*2];
bool pc[N*2];
int dep[N],fa[N],zon[N],sz[N],ld[N],xl[N],pos[N],top;
void dfs1(int c)
{
dep[c]=dep[fa[c]]+1;
sz[c]=1;
for(int t=fir[c];t;t=nes[t])
{
if(pc[t]) continue;
pc[t^1]=true;
fa[v[t]]=c;
dfs1(v[t]);
sz[c]+=sz[v[t]];
if(sz[v[t]]>sz[zon[c]]) zon[c]=v[t];
}
}
void dfs2(int c)
{
xl[++top]=c;
pos[c]=top;
ld[c]=c;
if(zon[fa[c]]==c) ld[c]=ld[fa[c]];
if(zon[c]) dfs2(zon[c]);
for(int t=fir[c];t;t=nes[t])
{
if(pc[t] || v[t]==zon[c]) continue;
dfs2(v[t]);
}
}
void make_tree(int c,int l,int r)
{
d[c].l=l;d[c].r=r;
if(l==r)
{
d[c].sum=1;
d[c].ls=d[c].rs=q[xl[l]];
d[c].lz=-1;
return;
}
int mid=(l+r)/2;
make_tree(c*2,l,mid);
make_tree(c*2+1,mid+1,r);
d[c].sum=d[c*2].sum+d[c*2+1].sum;
if(d[c*2].rs==d[c*2+1].ls) d[c].sum--;
d[c].ls=d[c*2].ls;
d[c].rs=d[c*2+1].rs;
d[c].lz=-1;
}
void push_down(int c)
{
if(d[c].lz==-1) return;
d[c].sum=1;
d[c].ls=d[c].rs=d[c].lz;
d[c*2].lz=d[c*2+1].lz=d[c].lz;
d[c].lz=-1;
return;
}
void change(int c)
{
if(a<=d[c].l && b>=d[c].r) {d[c].lz=z;return;}
push_down(c);
int mid=(d[c].l+d[c].r)/2;
if(a<=mid) change(c*2);
if(b>mid) change(c*2+1);
push_down(c*2);
push_down(c*2+1);
d[c].sum=d[c*2].sum+d[c*2+1].sum;
if(d[c*2].rs==d[c*2+1].ls) d[c].sum--;
d[c].ls=d[c*2].ls;
d[c].rs=d[c*2+1].rs;
d[c].lz=-1;
}
void lca_change(int x,int y)
{
while(ld[x]!=ld[y])
{
if(dep[ld[x]]<dep[ld[y]]) swap(x,y);
a=pos[ld[x]]; b=pos[x];
change(1);
x=fa[ld[x]];
}
if(dep[x]<dep[y]) swap(x,y);
a=pos[y]; b=pos[x];
change(1);
}
int query(int c,int &zs,int &ys)
{
push_down(c);
if(a<=d[c].l && b>=d[c].r)
{
zs=d[c].ls; ys=d[c].rs;
return d[c].sum;
}
int mid=(d[c].l+d[c].r)/2;
if(b<=mid) return query(c*2,zs,ys);
if(a>mid) return query(c*2+1,zs,ys);
int zz,yy,ans=0;
ans+=query(c*2,zs,yy);
ans+=query(c*2+1,zz,ys);
if(d[c*2].rs==d[c*2+1].ls) ans--;
return ans;
}
int lca_query(int x,int y)
{
int lx=x,ly=y;
while(ld[lx]!=ld[ly])
{
if(dep[ld[lx]]<dep[ld[ly]]) swap(lx,ly);
lx=fa[ld[lx]];
}
if(dep[lx]<dep[ly]) swap (lx,ly);
int lca=ly,ans=0;
int ls=-1,zs=-1,ys=-1,k1,k2;
while(ld[x]!=ld[lca])
{
a=pos[ld[x]];b=pos[x];
ans+=query(1,zs,ys);
if(ls==ys) ans--;
ls=zs;
x=fa[ld[x]];
}
a=pos[lca];b=pos[x];
ans+=query(1,zs,ys);
if(ls==ys) ans--;
k1=zs;
ls=zs=ys=-1;
while(ld[y]!=ld[lca])
{
a=pos[ld[y]];b=pos[y];
ans+=query(1,zs,ys);
if(ls==ys) ans--;
ls=zs;
y=fa[ld[y]];
}
if(y!=lca)
{
a=pos[lca]+1; b=pos[y];
ans+=query(1,zs,ys);
if(ls==ys) ans--;
}
k2=zs;
if(k1==k2) ans--;
return ans;
}
int main()
{
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++) scanf("%d",&q[i]);
for(int i=2;i<2*n;i+=2)
{
scanf("%d%d",&x,&y);
nes[i]=fir[x];
fir[x]=i; v[i]=y;
nes[i^1]=fir[y];
fir[y]=i^1; v[i^1]=x;
}
dfs1(1);dfs2(1);
make_tree(1,1,n);
for(int i=1;i<=m;i++)
{
scanf("%s",s);
if(s[0]=='Q')
{
scanf("%d%d",&x,&y);
printf("%d\n",lca_query(x,y));
}
else
{
scanf("%d%d%d",&x,&y,&z);
lca_change(x,y);
}
}
return 0;
}