Count on a tree II（树上莫队）

#include <cstdio>
#include <cstdlib>
#include <algorithm>
#include <vector>
#include <cmath>
using namespace std;
const int N=1e5+10,M=4e4+10;
int ans[N],an,s_d[M],s_u[M],u,v,n,m,d[M],cnt,tot,pos[N],st[M],ed[M],si,bl,fa[M],dep[M],b[N];
int top[M],son[M],siz[M];
vector <int> link[N];
struct node
{
    int l,r,id,lca;
    bool operator <(const node &o)const
    {
        return pos[l]^pos[o.l]?l<o.l:(pos[l]&1?r<o.r:r>o.r);
    }
}q[N];
struct mode
{
    int v,id;
    bool operator <(const mode &o)const
    {
        return v<o.v;
    }
}a[M];
void dfs(int u,int fx)
{
    st[u]=++cnt,b[cnt]=u;
    fa[u]=fx,dep[u]=dep[fx]+1;
    int size=link[u].size();
    for(int i=0;i<size;i++)
    {
        int v=link[u][i];
        if(v!=fx) dfs(v,u);
    }
    ed[u]=++cnt,b[cnt]=u;
}
void dfs1(int u)
{
    siz[u]=1;
    int size=link[u].size();
    for(int i=0;i<size;i++)
    {
        int v=link[u][i];
        if(v==fa[u]) continue;
        dfs1(v),siz[u]+=siz[v];
        if(son[u]==0 || siz[v]>siz[son[u]]) son[u]=v;
     }
}
void dfs2(int u)
{
    if(son[u]) top[son[u]]=top[u],dfs2(son[u]);
    else return ;
    int size=link[u].size();
    for(int i=0;i<size;i++)
    {
        int v=link[u][i];
        if(v==fa[u] || v==son[u]) continue;
        top[v]=v,dfs2(v);
    }
}
int Lca(int x,int y)
{
    int px=top[x],py=top[y];
    while(px!=py)
        if(dep[px]>dep[py])
            x=fa[px],px=top[x];
        else y=fa[py],py=top[y];
    if(dep[x]>dep[y]) return y;
    else return x;
}
void get(int x,int y)
{
    if(x==0) return;
    s_u[x]+=y;
    if(s_u[x]&1)
    {
        s_d[d[x]]++;
        if(s_d[d[x]]==1) an++;
    }
    else
    {
        s_d[d[x]]--;
        if(s_d[d[x]]==0) an--;
    }
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
        scanf("%d",&a[i].v),a[i].id=i;
    sort(a+1,a+1+n);
    for(int i=1;i<=n;i++)
    {
        if(i==1 || a[i].v!=a[i-1].v) cnt++;
        d[a[i].id]=cnt;
    }
    for(int i=1;i<n;i++)
    {
        scanf("%d%d",&u,&v);
        link[u].push_back(v);
        link[v].push_back(u);
    }
    cnt=0,dfs(1,0);
    dfs1(1),top[1]=1,dfs2(1);
    for(int i=0;i<m;i++)
    {
        scanf("%d%d",&u,&v);
        if(st[u]>st[v]) swap(u,v);
        int w=Lca(u,v);
        if(w==u) q[i].l=st[u],q[i].r=st[v],q[i].lca=0;
        else q[i].lca=d[w],q[i].l=ed[u],q[i].r=st[v];
        q[i].id=i;
    }
    n=cnt;
    si=sqrt(n);
    bl=(n+si-1)/si;
    for(int i=1;i<=bl;i++)
    {
        int z=min(n,i*si);
        for(int j=(i-1)*si+1;j<=z;j++)
            pos[j]=i;
    }
    sort(q,q+m);
    int nl=0,nr=0;
    for(int i=0;i<m;i++)
    {
        while(nl>q[i].l) get(b[--nl],1);
        while(nl<q[i].l) get(b[nl++],-1);
        while(nr>q[i].r) get(b[nr--],-1);
        while(nr<q[i].r) get(b[++nr],1);
        ans[q[i].id]=an;
        if(q[i].lca) ans[q[i].id]+=(s_d[q[i].lca]==0);
    }
    for(int i=0;i<m;i++)
        printf("%d\n",ans[i]);
    return 0;
}