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
注意兩點:線段樹區間要記錄最左段顏色、最右段顏色和該區間顏色段數,用來自底向上更新;
再一個是求路徑a~b上的顏色段數時,拆開的區間相接的位置顏色相同便要合併爲爲一個顏色段,具體見代碼。
//#pragma comment(linker, "/STACK:1024000000,1024000000")
#include<iostream>
#include<stdio.h>
#include<math.h>
#include <string>
#include<string.h>
#include<map>
#include<queue>
#include<set>
#include<utility>
#include<vector>
#include<algorithm>
#include<stdlib.h>
using namespace std;
#define eps 1e-8
#define pii pair<int,int>
#define inf 0x3f3f3f3f
#define rd(x) scanf("%d",&x)
#define rd2(x,y) scanf("%d%d",&x,&y)
#define ll long long int
#define mod 1000000007
#define maxn 100050
#define maxm 1000005
struct edge{
int to,next;
}edge[maxn*2];
int head[maxn],tot;
int top[maxn];//v所在重鏈的頂端節點
int fa[maxn];
int deep[maxn];
int num[maxn];//表示以v爲根的子樹的節點數
int p[maxn];//表示v在線段樹的位置
int fp[maxn];//
int son[maxn];//重兒子
int pos;
int v[maxn];
int maxx(int a,int b){return a>b?a:b;}
int mixx(int a,int b){return a<b?a:b;}
void init(){
tot=0;pos=1;
memset(head,-1,sizeof(head));
memset(son,-1,sizeof(son));
}
void addedge(int u,int v){
edge[tot].to=v;edge[tot].next=head[u];head[u]=tot++;
}
//第一遍dfs求fa,deep,num,son
void dfs1(int u,int pre,int d){
deep[u]=d;
fa[u]=pre;
num[u]=1;
for(int i=head[u];i!=-1;i=edge[i].next){
int v=edge[i].to;
if(v!=pre){
dfs1(v,u,d+1);
num[u]+=num[v];
if(son[u]==-1||num[v]>num[son[u]]) son[u]=v;
}
}
}
//第二遍dfs求出top和p
void getpos(int u,int sp){
top[u]=sp;
p[u]=pos++;
fp[p[u]]=u;
if(son[u]==-1) return;
getpos(son[u],sp);
for(int i=head[u];i!=-1;i=edge[i].next){
int v=edge[i].to;
if(v!=son[u]&&v!=fa[u]) getpos(v,v);
}
}
struct node{
int l,r,col,lc,rc,nc;//
}tree[maxn*3];
void push_down(int i){//向下更新顏色更改
if(tree[i].l==tree[i].r) {
v[tree[i].l]=tree[i].col;
return;
}
if(tree[i].col!=-1){
tree[i<<1].col=tree[(i<<1)|1].col=tree[i].col;
tree[i<<1].lc=tree[i<<1].rc=tree[i].col;
tree[(i<<1)|1].lc=tree[(i<<1)|1].rc=tree[i].col;
tree[i<<1].nc=tree[(i<<1)|1].nc=1;
tree[i].col=-1;
}
}
void push_up(int i){
tree[i].nc=tree[i<<1].nc+tree[(i<<1)|1].nc;
if(tree[i<<1].rc==tree[(i<<1)|1].lc) tree[i].nc--;
tree[i].lc=tree[i<<1].lc;tree[i].rc=tree[(i<<1)|1].rc;
}
void build(int i,int l,int r){
tree[i].l=l;tree[i].r=r;
tree[i].col=-1;
if(l==r) {
tree[i].col=v[tree[i].l];
tree[i].lc=tree[i].rc=tree[i].col;
tree[i].nc=1;
return;
}
int mid=(l+r)>>1;
build(i<<1,l,mid);
build((i<<1)|1,mid+1,r);
push_up(i);
}
int query(int i,int l,int r){//詢問l~r區間有多少顏色區間
push_down(i);
if(tree[i].l==l&&tree[i].r==r) return tree[i].nc;
int mid=(tree[i].l+tree[i].r)>>1;
if(r<=mid) return query(i<<1,l,r);
else if(l>mid) return query((i<<1)|1,l,r);
else{
if(tree[i<<1].rc==tree[(i<<1)|1].lc) return query(i<<1,l,mid)+query((i<<1)|1,mid+1,r)-1;
else return query(i<<1,l,mid)+query((i<<1)|1,mid+1,r);
}
push_up(i);
}
void update(int i,int l,int r,int val){//將l~r區間的顏色改爲val
if(tree[i].l==l&&tree[i].r==r){
tree[i].col=val;
tree[i].lc=tree[i].rc=val;
tree[i].nc=1;
if(tree[i].l==tree[i].r) v[tree[i].l]=tree[i].col;
return;
}
push_down(i);
int mid=(tree[i].l+tree[i].r)>>1;
if(mid>=r) update(i<<1,l,r,val);
else if(l>mid) update((i<<1)|1,l,r,val);
else{
update(i<<1,l,mid,val);
update((i<<1)|1,mid+1,r,val);
}
push_up(i);
}
void change(int u,int v,int val){//u->v的點變爲val
int f1=top[u],f2=top[v];
// int tmp=-1000000000;
while(f1!=f2){
if(deep[f1]<deep[f2]){
swap(f1,f2);
swap(u,v);
}
update(1,p[f1],p[u],val);
u=fa[f1];f1=top[u];
}
// if(u==v) return ;
if(deep[u]>deep[v]) swap(u,v);
//return maxx(tmp,query(1,p[son[u]],p[v]));
update(1,p[u],p[v],val);
}
int getcol(int i,int x){//獲取x點的顏色
if(tree[i].l==tree[i].r) return tree[i].col;
int mid=(tree[i].l+tree[i].r)>>1;
push_down(i);
if(x<=mid) return getcol(i<<1,x);
else return getcol((i<<1)|1,x);
push_up(i);
}
int query(int x,int y){//詢問樹上x-y路徑上的顏色區間數
int f1=top[x],f2=top[y];
// int tmp=-1000000000;
int res=0;
while(f1!=f2){
if(deep[f1]<deep[f2]){
swap(f1,f2);
swap(x,y);
}
res+=query(1,p[f1],p[x]);
if(getcol(1,p[f1])==getcol(1,p[fa[f1]])) res--;//若前後區間相交處顏色相同併爲一個區間
x=fa[f1];f1=top[x];
}
// if(u==v) return ;
if(deep[x]>deep[y]) swap(x,y);
//return maxx(tmp,query(1,p[son[u]],p[v]));
res+=query(1,p[x],p[y]);
return res;
}
int t,n,m,pp,x,y,z;
int vv[maxn];
char s[10];
int main()
{
while(~scanf("%d%d",&n,&m)){
for(int i=1;i<=n;i++) rd(vv[i]);
init();
for(int i=1;i<n;i++){
rd2(x,y);
addedge(x,y);addedge(y,x);
}
dfs1(1,0,0);
getpos(1,1);
for(int i=1;i<=n;i++) v[p[i]]=vv[i];
build(1,1,n);
for(int i=1;i<=m;i++){
scanf("%s",s);
if(s[0]=='C'){
scanf("%d%d%d",&x,&y,&z);
change(x,y,z);
}else{
rd2(x,y);
printf("%d\n",query(x,y));
}
}
}
return 0;
}