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

树上莫队模板题。

使用欧拉序将树上路径转化为普通区间。

之后莫队维护即可。不要忘记特判LCA

 #include<iostream>
 #include<cstdio>
 #include<cstring>
 #include<cmath>
 #include<algorithm>
 #define N 200005
 using namespace std;
 int read()
 {
     int x=,f=;char ch=getchar();
     while(ch<''||ch>''){if(ch=='-')f=-;ch=getchar();}
     while(ch>=''&&ch<=''){x=(x<<)+(x<<)+(ch^);ch=getchar();}
     return x*f;
 }
 int n,m,val[N],inp[N],cnt[N],v[N],nxt[N],head[N],cntt,tot,siz,ord[N],ncnt,fir[N],lst[N];
 int fa[N][],dep[N],now,l=,r=,ans[N],vis[N];
 struct node
 {
     int l,r,x,lca;
 }q[N];
 bool cmp(node a,node b)
 {
     if(a.l/siz!=b.l/siz)return a.l/siz<b.l/siz;
     return a.r<b.r;
 }
 void add(int a,int b)
 {
     v[++cntt]=b;
     nxt[cntt]=head[a];
     head[a]=cntt;
 }
 void dfs1(int x,int f)
 {
     dep[x]=dep[f]+;
     for(int i=;i<=;i++)fa[x][i+]=fa[fa[x][i]][i];
     for(int i=head[x];i;i=nxt[i])
     {
         if(v[i]==f)continue;
         fa[v[i]][]=x;
         dfs1(v[i],x);
     }
 }
 int lca(int x,int y)
 {
     if(dep[x]<dep[y])swap(x,y);
     for(int i=;i>=;i--)
     {
         if(dep[fa[x][i]]>=dep[y])x=fa[x][i];
         if(x==y)return x;
     }
     for(int i=;i>=;i--)
     {
         if(fa[x][i]!=fa[y][i])x=fa[x][i],y=fa[y][i];
     }
     return fa[x][];
 }
 void dfs2(int x,int f)
 {
     ord[++ncnt]=x;
     fir[x]=ncnt;
     for(int i=head[x];i;i=nxt[i])
     {
         if(v[i]==f)continue;
         dfs2(v[i],x);
     }
     ord[++ncnt]=x;
     lst[x]=ncnt;
 }
 void work(int pos)
 {
     if(vis[pos])now-=!--cnt[val[pos]];
     else now+=!cnt[val[pos]]++;
     vis[pos]^=;
 }
 int main()
 {
     n=read();m=read();
     for(int i=;i<=n;i++)val[i]=inp[i]=read();
     sort(inp+,inp+n+);
     tot=unique(inp+,inp+n+)-inp-;
     for(int i=;i<=n;i++)val[i]=lower_bound(inp+,inp+tot+,val[i])-inp;
     for(int x,y,i=;i<n;i++)
     {
         x=read();y=read();
         add(x,y);add(y,x);
     }
     dfs1(,);dfs2(,);
     for(int i=;i<=m;i++)
     {
         int ll=read(),rr=read(),LCA=lca(ll,rr);
         if(fir[ll]>fir[rr])swap(ll,rr);
         if(ll==LCA)
         {
             q[i].l=fir[ll];
             q[i].r=fir[rr];
         }
         else
         {
             q[i].l=lst[ll];
             q[i].r=fir[rr];
             q[i].lca=LCA;
         }
         q[i].x=i;
     }
     siz=sqrt(ncnt);
     sort(q+,q+m+,cmp);
     for(int i=;i<=m;i++)
     {
         int ll=q[i].l,rr=q[i].r,Lca=q[i].lca;
         while(l<ll)work(ord[l++]);
         while(l>ll)work(ord[--l]);
         while(r<rr)work(ord[++r]);
         while(r>rr)work(ord[r--]);
         if(Lca)work(Lca);
         ans[q[i].x]=now;
         if(Lca)work(Lca);
     }
     for(int i=;i<=m;i++)printf("%d\n",ans[i]);
     return ;
 }