题目描述

在这里插入图片描述

解析

用线段树维护颜色序列个数、最左颜色与最右颜色
合并时如果左儿子的最右颜色等于右儿子的最左颜色，就把加和-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*/