传送门
首先,对hash学姐对本题拔刀相助的debug行为表示衷心的感谢
题目描述
解析
用线段树维护颜色序列个数、最左颜色与最右颜色
合并时如果左儿子的最右颜色等于右儿子的最左颜色,就把加和-1
在加上树剖的基本操作即可
几点注意事项:
1.从重链头往父亲跳的时候要判断一下,如果链头与父亲同色,需要减一
2.线段树区间查询合并答案也要注意两边如果颜色相同要减一!(我就因为这个调了一个多点。。。)
代码
#include<bits/stdc++.h>
using namespace std;
const int N=4e5+100;
#define ll long long
#define ls k<<1
#define rs k<<1|1
int fa[N],dep[N],size[N];
int dfs[N],pos[N],tot;
int top[N],hson[N];
ll b[N],a[N];
ll add[4*N],col[4*N],le[4*N],ri[4*N];
int n,m,root;
struct node{
int to,nxt;
}p[N*2];
int fi[N],cnt=-1;
void addline(int x,int y){
p[++cnt]=(node){y,fi[x]};
fi[x]=cnt;
}
void merge(int k){
le[k]=le[ls];
ri[k]=ri[rs];
col[k]=col[ls]+col[rs];
if(ri[ls]==le[rs]&&col[k]>1) col[k]--;
}
void Add(int k,int l,int r,int v){
add[k] = v;
ri[k]=le[k]=v;
col[k] = 1;
}
void pushdown(int k,int l,int r,int mid){
if(add[k]==-1) return;
Add(2*k,l,mid,add[k]);
Add(2*k+1,mid+1,r,add[k]);
add[k] = -1;
}
long long longquery(int k,int l,int r,int x,int y){
if(x<=l&&r<=y) return col[k];
int mid=(l+r) >> 1;
pushdown(k,l,r,mid);
long long res=0;
if(x<=mid) res += longquery(2*k,l,mid,x,y);
if(y>mid) res += longquery(2*k+1,mid+1,r,x,y);
if(x<=mid&&y>=mid+1&&ri[ls]==le[rs]) res--;
merge(k);
return res;
}
void longchange(int k,int l,int r,int x,int y,int v){
if(x<=l&&r<=y){
Add(k,l,r,v);
return;
}
int mid = (l+r) >> 1;
pushdown(k,l,r,mid);
if(x<=mid) longchange(2*k,l,mid,x,y,v);
if(y>mid) longchange(2*k+1,mid+1,r,x,y,v);
merge(k);
return;
}
void build(int k,int l,int r){
if(l==r){
col[k]=1;ri[k]=le[k]=a[l];
// printf("pl=%d le=%d\n",l,a[l]);
return;
}
int mid=(l+r)/2;
build(2*k,l,mid);
build(2*k+1,mid+1,r);
merge(k);
return;
}
int dotask(int k,int l,int r,int x){
if(l==r){
return ri[k];
}
int mid=l+r>>1;
pushdown(k,l,r,mid);
if(x<=mid) return dotask(ls,l,mid,x);
else return dotask(rs,mid+1,r,x);
merge(k);
}
void dfs1(int x,int f){
size[x]=1;
for(int i=fi[x];~i;i=p[i].nxt){
int to=p[i].to;
if(to==f) continue;
dep[to]=dep[x]+1;
fa[to]=x;
dfs1(to,x);
size[x]+=size[to];
if(!hson[x]||size[to]>size[hson[x]]) hson[x]=to;
}
return;
}
void dfs2(int x,int tp){
top[x]=tp;
dfs[++tot]=x;
pos[x]=tot;
a[tot]=b[x];
if(!hson[x]) return;
dfs2(hson[x],tp);
for(int i=fi[x];~i;i=p[i].nxt){
int to=p[i].to;
if(to==fa[x]) continue;
if(to==hson[x]) continue;
else dfs2(to,to);
}
return;
}
void roadchange(int x,int y,int v){
while(top[x]!=top[y]){
if(dep[top[x]]<dep[top[y]]) swap(x,y);
longchange(1,1,tot,pos[top[x]],pos[x],v);
x=fa[top[x]];
}
if(dep[x]<dep[y]) swap(x,y);
longchange(1,1,tot,pos[y],pos[x],v);
return;
}
ll roadask(int x,int y){
ll ans=0;
while(top[x]!=top[y]){
if(dep[top[x]]<dep[top[y]]) swap(x,y);
ans+=longquery(1,1,tot,pos[top[x]],pos[x]);
// printf("+=%d %d\n",pos[top[x]],pos[x]);
// printf("x=%d top=%d fa=%d\n",x,top[x],fa[top[x]]);
// printf("col1=%d col2=%d\n",dotask(1,1,tot,pos[top[x]]),dotask(1,1,tot,pos[fa[top[x]]]));
if(dotask(1,1,tot,pos[top[x]])==dotask(1,1,tot,pos[fa[top[x]]])) ans--;
x=fa[top[x]];
}
if(dep[x]<dep[y]) swap(x,y);
ans+=longquery(1,1,tot,pos[y],pos[x]);
// printf("+=%d %d\n",pos[y],pos[x]);
return ans;
}
void print_dfs(){
for(int i=1;i<=tot;i++) printf("i=%d pos=%d size=%d hson=%d top=%d\n",i,pos[i],size[i],hson[i],top[i]);
}
int main(){
memset(fi,-1,sizeof(fi));
memset(add,-1,sizeof(add));
scanf("%d%d",&n,&m);
int u,v,w;
for(int i=1;i<=n;i++) scanf("%lld",&b[i]);
for(int i=1;i<n;i++){
scanf("%d%d",&u,&v);
addline(u,v);
addline(v,u);
}
dfs1(1,0);
dfs2(1,1);
build(1,1,tot);
// print_dfs();
for(int i=1;i<=m;i++){
char flag;
scanf(" %c ",&flag);
if(flag=='C'){
scanf("%d%d%d",&u,&v,&w);
roadchange(u,v,w);
}
// else if(flag=='a'){
// scanf("%d",&u);
// printf("%d\n",dotask(1,1,tot,pos[u]));
// }
else{
scanf("%d%d",&u,&v);
printf("%lld\n",roadask(u,v));
}
}
}
/*
6 155
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
*/