[树上莫队] SPOJ COT2 Count on a tree II

题意

给定n个节点的一棵树，每个节点上都有一个数字。
m次询问，没次求u到v的路径上有几种不同的数字。

题解

树上莫队经典题。
首先要会树分块，见bzoj1086: [SCOI2005]王室联邦。
与普通莫队类似，记dfn[x]为x的入栈时间戳，我们按（belong[U], dfn[V]）排序所有询问。
显然分块完后属于同一块的最远点对距离不超过块的大小，
而且考虑括号序列，dfn[V]递增的话一步一步走过去不会超过2n部。
因此复杂度和普通莫队类似。
如何从当前状态转移到下一个呢？可以发现对于LCA那个点有点特殊。
对于这题，如果我们得到除了lca那个点其他的信息，只需要加上lca那一个点即可得到答案。
设S(u,v)为从u到v的路径除去LCA那个点的信息,T(x)表示根到x的信息。
如果我们把v改为v2 ，考虑变化了哪些(下面的加是指类似异或的运算，即有2个相同元素就约掉)：
$S(u,v) + S(u,v_2)$
$=( T(u) + T(v) + (T(u) + T(v_2) )$
$=( T(v) + T(v_2) )=S(v,v_2)$
也就是说，只要把v到v2的路径上除了lca(v,v2)的点的存在性取反即可。
复杂度$O(n\sqrt n)$

#include<cstdio>
#include<cmath>
#include<algorithm>
using namespace std;
const int maxn=40005, maxe=80005, maxq=100005;
int n,m,c[maxn],b[maxn],sum[maxn],ans,dfn[maxn],Tim,G,blk,blg[maxn],stk[maxn],top,id_c[maxn];
int son[maxe],fir[maxn],nxt[maxe],tot;
bool vis[maxn];
int dfs(int x,int pre){
    dfn[x]=++Tim; int bottom=top;
    for(int j=fir[x];j;j=nxt[j]) if(son[j]!=pre){
        dfs(son[j],x);
        if(top-bottom>=blk){
            G++;
            while(top!=bottom) blg[stk[top--]]=G;
        }
    }
    stk[++top]=x;
}
struct data{ int x,y,id,res; } q[maxq];
bool my_cmp1(const data &A,const data &B){
    if(blg[A.x]==blg[B.x]) return dfn[A.y]<dfn[B.y];
    return blg[A.x]<blg[B.x];
}
bool my_cmp2(const data &A,const data &B){ return A.id<B.id; }
void add(int x,int y){
    son[++tot]=y; nxt[tot]=fir[x]; fir[x]=tot;
}
int anc[maxn][20],dep[maxn];
void dfs_LCA(int x,int pre){
    anc[x][0]=pre; for(int i=1;i<=17;i++) anc[x][i]=anc[anc[x][i-1]][i-1];
    for(int j=fir[x];j;j=nxt[j]) if(son[j]!=pre) dep[son[j]]=dep[x]+1, dfs_LCA(son[j],x);
}
int LCA(int x,int y){
    if(dep[x]<dep[y]) swap(x,y);
    for(int j=17;j>=0;j--) if(dep[anc[x][j]]>=dep[y]) x=anc[x][j];
    if(x==y) return x;
    for(int j=17;j>=0;j--) if(anc[x][j]!=anc[y][j]) x=anc[x][j], y=anc[y][j];
    return anc[x][0];
}
int find(int x){
    int L=1,R=b[0];
    while(L<=R){
        int mid=(L+R)>>1;
        if(b[mid]==x) return mid;
        if(b[mid]<x) L=mid+1; else R=mid-1;
    }
}
void rev(int x){
    if(vis[x]) ans=(--sum[id_c[x]])?ans:ans-1;
          else ans=(sum[id_c[x]]++)?ans:ans+1;
    vis[x]^=1; 
}
void work(int x,int y){
    while(x!=y){
        if(dep[x]<dep[y]) swap(x,y);
        rev(x); x=anc[x][0];
    }
}
int main(){
    freopen("cot2.in","r",stdin);
    freopen("cot2.out","w",stdout);
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++) scanf("%d",&c[i]), b[++b[0]]=c[i];
    for(int i=1;i<=n-1;i++){
        int x,y; scanf("%d%d",&x,&y);
        add(x,y); add(y,x);
    }
    dfs_LCA(1,1);
    blk=sqrt(n); dfs(1,0);
    for(int i=1;i<=m;i++){
        scanf("%d%d",&q[i].x,&q[i].y); q[i].id=i;
        if(dfn[q[i].x]>dfn[q[i].y]) swap(q[i].x,q[i].y);
    }
    sort(b+1,b+1+b[0]); b[0]=unique(b+1,b+1+b[0])-(b+1);
    sort(q+1,q+1+m,my_cmp1);
    for(int i=1;i<=n;i++) id_c[i]=find(c[i]);
    q[0].x=q[0].y=1;
    for(int i=1;i<=m;i++){
        work(q[i-1].x,q[i].x);
        work(q[i-1].y,q[i].y);
        int _lca=LCA(q[i].x,q[i].y);
        rev(_lca); q[i].res=ans; rev(_lca);
    }
    sort(q+1,q+1+m,my_cmp2);
    for(int i=1;i<=m;i++) printf("%d\n",q[i].res);
    return 0;
}