题意
给定一棵有n个节点的无根树和m个操作,操作有2类:
1、将节点a到节点b路径上所有点都染成颜色c;
2、询问节点a到节点b路径上的颜色段数量(连续相同颜色被认为是同一段),如“112221”由3段组成:“11”、“222”和“1”。
请你写一个程序依次完成这m个操作。
应该是我太水了来着…?打了一小时调了一天,主要就是注意合并时判断中间部分的端点颜色,如果一样的话统计时就要减1。然后在get函数中也一样,接到上一段的时候要判断端点的颜色,这题还是很有价值的…然后那个数组ni,就是逆的意思,表示线段树中的结点的原节点…
#include <cstdio>
#include <algorithm>
#define Rep(i,s,t) for(int i=s;i<=t;i++)
#define For(i,s,t) for(int i=s;i;i=t)
using namespace std;
const int maxx = 100000 + 25;
int head[maxx],to[maxx<<1],nxt[maxx<<1],a[maxx],ni[maxx];
int top[maxx],dpt[maxx],son[maxx],ftr[maxx],rnk[maxx],size[maxx];
int T[maxx<<2],col[maxx<<2],Add[maxx<<2],lc[maxx<<2],rc[maxx<<2];
int n,m,num,cnt,x,y,z;
char f[5];
namespace Y{
void Ins(int x,int y){
to[++num] = y;nxt[num] = head[x];head[x] = num;
to[++num] = x;nxt[num] = head[y];head[y] = num;
}
void Dfs1(int x){
size[x] = 1;
For(i , head[x] , nxt[i]){
int now = to[i];if(now == ftr[x]) continue;
ftr[now] = x;dpt[now] = dpt[x] + 1;
Dfs1(now);size[x] += size[now];
if(size[now] > size[son[x]]) son[x] = now;
}
}
void Dfs2(int x,int brn){
rnk[x] = ++cnt;top[x] = brn;
ni[cnt] = x;
if(son[x]) Dfs2(son[x],brn);
For(i , head[x] , nxt[i])
if(to[i]!=ftr[x] && to[i]!=son[x])
Dfs2(to[i],to[i]);
}
void upt(int i){
lc[i] = lc[i<<1];rc[i] = rc[i<<1|1];
T[i] = T[i<<1] + T[i<<1|1];
if(rc[i<<1] == lc[i<<1|1]) T[i]--;
}
void pushdown(int i){
Add[i<<1] = Add[i<<1|1] = Add[i];
lc[i<<1] = rc[i<<1] = lc[i<<1|1] = rc[i<<1|1] = Add[i];
T[i<<1] = T[i<<1|1] = 1;Add[i] = 0;
}
void Build(int i,int l,int r){
if(l == r) {lc[i] = rc[i] = a[ni[l]];T[i] = 1;return;}
int mid = (l+r) >> 1;
Build(i<<1,l,mid);Build(i<<1|1,mid+1,r);
upt(i);
}
void modify(int i,int x,int y,int l,int r,int k){
if(x <= l && r <= y){T[i]=1;lc[i]=rc[i]=k;Add[i]=k;return;}
if(Add[i]) pushdown(i);int mid = (l+r) >> 1;
if(x <= mid) modify(i<<1,x,y,l,mid,k);
if(y > mid) modify(i<<1|1,x,y,mid+1,r,k);
upt(i);
}
int Query(int i,int x,int y,int l,int r){
if(x <= l && r <= y) return T[i];int ans = 0;
if(Add[i]) pushdown(i);int mid = (l+r) >> 1;
if(x <= mid) ans += Query(i<<1,x,y,l,mid);
if(y > mid) ans += Query(i<<1|1,x,y,mid+1,r);
if(x <= mid && y > mid && rc[i<<1] == lc[i<<1|1]) return ans-1;
return ans;
}
void update(int x,int y,int k){
while(top[x] != top[y]){
if(dpt[top[x]] > dpt[top[y]]) x^=y^=x^=y;
modify(1,rnk[top[y]],rnk[y],1,n,k);
y = ftr[top[y]];
}
if(rnk[x] > rnk[y]) x^=y^=x^=y;
modify(1,rnk[x],rnk[y],1,n,k);
}
int col(int i,int pos,int l,int r){
if(l == r) return lc[i];
if(Add[i]) pushdown(i);
int mid = (l+r) >>1;
if(pos <= mid) return col(i<<1,pos,l,mid);
else return col(i<<1|1,pos,mid+1,r);
}
int get(int x,int y){
int ans = 0;
while(top[x] != top[y]){
if(dpt[top[x]] > dpt[top[y]]) x^=y^=x^=y;
int a = col(1,rnk[top[y]],1,n),b = col(1,rnk[ftr[top[y]]],1,n);
ans += Query(1,rnk[top[y]],rnk[y],1,n);
if(a == b) ans -- ;
y = ftr[top[y]];
}
if(rnk[x] > rnk[y]) x^=y^=x^=y;
ans += Query(1,rnk[x],rnk[y],1,n);
return ans;
}
}
using namespace Y;
int main(){
scanf("%d%d",&n,&m);Rep(i,1,n) scanf("%d",&a[i]);
Rep(i,1,n-1) scanf("%d%d",&x,&y),Ins(x,y);
Dfs1(1);Dfs2(1,1);Build(1,1,n);
while( m-- ){
scanf("%s",f);
if(f[0] == 'Q') scanf("%d%d",&x,&y),printf("%d\n",get(x,y));
if(f[0] == 'C') scanf("%d%d%d",&x,&y,&z),update(x,y,z);
}
return 0;
}