spoj_cot2 Count on a tree II(树上莫队+离散化)

题意：给两个点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;
}