又是一道一眼看去就是樹鏈剖分的題。
仍然是線段樹，維護一段區間內顏色段的數目。
此題難點是線段樹的寫法以及邊界判斷：若兩段區間相鄰且處於左邊的區間右端點和右邊的區間左端點被塗有相同的顏色，則這兩段區間總顏色段的數量爲各自的數量之和 -1 。在對樹上兩點間進行詢問時，對於兩條重鏈之間的交界處同樣需要注意這一問題！
樹鏈剖分部分請參看【算法雜談_02】樹鏈剖分
[SDOI2011]染色 C++代碼實現
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
#define lson pos<<1
#define rson pos<<1|1
#define N 300005
int n,a[N],ans[N];
int cnt,head[N],next[2*N],to[2*N];
int now,size[N],deep[N],fa[N],son[N],p_id[N],id_p[N],top[N];
int f[4*N];
void add(int x,int y)
{
to[++cnt]=y;
next[cnt]=head[x];
head[x]=cnt;
}
void dfs(int x,int d)
{
size[x]=1;deep[x]=d;
for(int y,i=head[x];i;i=next[i])
if((y=to[i])!=fa[x])
{
fa[y]=x;
dfs(y,d+1);
if(size[y]>size[son[x]])
son[x]=y;
size[x]+=size[y];
}
}
void create(int x,int d)
{
p_id[x]=++now;id_p[now]=x;
id_p[now]=x;
top[x]=d;
if(son[x])
create(son[x],d);
for(int y,i=head[x];i;i=next[i])
if((y=to[i])!=fa[x]&&y!=son[x])
create(y,y);
}
void lazy(int pos)
{
f[lson]+=f[pos];
f[rson]+=f[pos];
f[pos]=0;
}
void fix(int pos,int l,int r,int x,int y)
{
if(x<=l&&r<=y){ f[pos]++; return; }
if(f[pos]) lazy(pos);
int mid=(l+r)>>1;
if(y<=mid)
fix(lson,l,mid,x,y);
else if(x>mid)
fix(rson,mid+1,r,x,y);
else
fix(lson,l,mid,x,y),
fix(rson,mid+1,r,x,y);
}
void getans(int pos,int l,int r)
{
if(l==r){ ans[id_p[l]]=f[pos]; return; }
if(f[pos]) lazy(pos);
int mid=(l+r)>>1;
getans(lson,l,mid);
getans(rson,mid+1,r);
}
void work(int x,int y)
{
int f1=top[x],f2=top[y];
while(f1!=f2)
{
if(deep[f1]<deep[f2])
swap(x,y),swap(f1,f2);
fix(1,1,n,p_id[f1],p_id[x]);
x=fa[f1];
f1=top[x];
}
if(deep[x]>deep[y])
swap(x,y);
fix(1,1,n,p_id[x],p_id[y]);
}
int main()
{
cin>>n;
for(int i=1;i<=n;i++)
scanf("%d",&a[i]);
for(int x,y,i=1;i<n;i++)
scanf("%d%d",&x,&y),
add(x,y),add(y,x);
dfs(1,1);
create(1,1);
for(int i=2;i<=n;i++)
work(a[i],a[i-1]);
getans(1,1,n);
for(int i=1;i<=n;i++)
printf("%d\n",i==a[1]?ans[i]:ans[i]-1);
return 0;
}