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 100005using 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;}