题意:给两个点u,v,询问两点路径上有多少种不同的权值。树上莫队。然后权值要离散化。树上区间转移呢,基本就是这样的:记上次询问的两点为lastu,lastv,现在询问的两点为u,v,那么对于路径(lastu,u)上的所有点的存在性取反(就是原来答案里有就删掉,没有就加上),对于(lastv,v)同理。但是因为还有讨厌的lca的存在,所以我们一直都没把lca放在答案里。最后记答案时单独处理lca。具体证明去看别人的博客啦~然后这题有个特别坑的地方:n不4000!!!
因为这个才TLE的!!我一直以为是自己的常数太大了,才过不去的。。没想到,手抖多打了个0,就过去了。。真是太讨厌了!!害我调了那么久。。怀疑自己是不是自带巨大常数不适合学习oi。。心痛!!!亲测开到5000就够了。。另外还有一点,在处理询问的时候,按所属块小的为左端点比按dfn小的为左端点要快。。不知有没有神犇能给我讲解一下这个分块究竟是怎么算最优的。。
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
#define N 50005
#define M 100005
using namespace std;
int n,m,a[N],aa[N],h[N],num=0,block,belong[N],f[N],ans=0,Log[N];
int dep[N],fa[N][17],dfn[N],dfn_num=0,block_num=0,ANS[N];
bool vis[N];
struct edge{
int to,next;
}data[N<<1];
struct query{
int l,r,id;
}q[M];
inline int swap(int &x,int &y){x^=y;y^=x;x^=y;}
inline int read(){
int x=0,f=1;char ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-') f=-1;ch=getchar();}
while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
return x*f;
}
int stack[N],top=0;
int dfs(int x){
int size=0;dfn[x]=++dfn_num;
for(int i=1;i<=Log[dep[x]];++i)//顺便把lca预处理了
fa[x][i]=fa[fa[x][i-1]][i-1];
for(int i=h[x];i;i=data[i].next){
int y=data[i].to;
if(y==fa[x][0]) continue;
fa[y][0]=x;dep[y]=dep[x]+1;
size+=dfs(y);
if(size>=block){
block_num++;
for(int i=1;i<=size;++i)
belong[stack[top--]]=block_num;
size=0;
}
}
stack[++top]=x;
return size+1;
}
inline bool cmp(query x,query y){
return belong[x.l]==belong[y.l]?dfn[x.r]<dfn[y.r]:belong[x.l]<belong[y.l];
}
inline int lca(int x,int y){
if(dep[x]<dep[y]) swap(x,y);
int d=dep[x]-dep[y];
for(int i=0;i<=Log[d];++i)
if((1<<i)&d) x=fa[x][i];
if(x==y) return x;
for(int i=Log[n];i>=0;--i)
if(fa[x][i]!=fa[y][i]) x=fa[x][i],y=fa[y][i];
return fa[x][0];
}
inline void reverse(int x){//对存在性取反
if(vis[x]){f[a[x]]--;if(!f[a[x]]) ans--;}
else{f[a[x]]++;if(f[a[x]]==1) ans++;}
vis[x]^=1;
}
inline void solve(int x,int y){
while(x!=y){
if(dep[x]>dep[y]) reverse(x),x=fa[x][0];
else reverse(y),y=fa[y][0];
}
}
int main(){
// freopen("a.in","r",stdin);
n=read();m=read();block=sqrt(n);
Log[0]=-1;
for(int i=1;i<=n;++i) Log[i]=Log[i>>1]+1;
for(int i=1;i<=n;++i) a[i]=read(),aa[i]=a[i];
sort(aa+1,aa+n+1);aa[0]=unique(aa+1,aa+n+1)-aa-1;
for(int i=1;i<=n;++i) a[i]=lower_bound(aa+1,aa+aa[0]+1,a[i])-aa;
for(int i=1;i<n;++i){
int x=read();int y=read();
data[++num].to=y;data[num].next=h[x];h[x]=num;
data[++num].to=x;data[num].next=h[y];h[y]=num;
}dfs(1);block_num++;
while(top) belong[stack[top--]]=block_num;
for(int i=1;i<=m;++i){
q[i].l=read();q[i].r=read();q[i].id=i;
//经试验,让l所属块更大会更快
if(belong[q[i].l]>belong[q[i].r]) swap(q[i].l,q[i].r);
}
sort(q+1,q+m+1,cmp);
int t=lca(q[1].l,q[1].r);
solve(q[1].l,q[1].r);
// reverse(t);
ANS[q[1].id]=ans+(!f[a[t]]);
// reverse(t);
for(int i=2;i<=m;++i){
solve(q[i-1].l,q[i].l);solve(q[i-1].r,q[i].r);
t=lca(q[i].l,q[i].r);
ANS[q[i].id]=ans+(!f[a[t]]);//lca单独算
}
for(int i=1;i<=m;++i)
printf("%d\n",ANS[i]);
return 0;
}