树上莫队——Tree and Queries

题解：

求出dfs序，然后树上莫队就可以出来了。

#include <bits/stdc++.h>
//#define int long long
typedef long long ll;
#define lowbit(x) x&(-x)
using namespace std;
const int N=2e5+10;
int n,m,c[N],ne[N<<1],head[N],e[N<<1],cnt,tot,in[N],out[N],dfn[N],block,b[N],con[N],sum[N];
void add(int a,int b){
    e[cnt]=b,ne[cnt]=head[a],head[a]=cnt++;
}
void dfs(int u,int fa){
    dfn[++tot]=u,in[u]=tot;
    for(int i=head[u];~i;i=ne[i]){
        int j=e[i];
        if(j!=fa) dfs(j,u);
    }
    out[u]=tot;
}
struct Node
{
    int l,r,k,id;
}node[N];
void del(int x){
    sum[con[c[x]]]--;
    con[c[x]]--;
}
void acc(int x){
    con[c[x]]++;
    sum[con[c[x]]]++;
}
int res[N];
int cmp(Node x,Node y)
{
   return b[x.l]==b[y.l]?x.r<y.r:x.l<y.l;
}
signed main()
{
    memset(head,-1,sizeof head);
    scanf("%d%d",&n,&m);
    block=sqrt(n);
    for(int i=1;i<=n;i++ ) scanf("%d",&c[i]),b[i]=(i-1)/block+1;
    for(int i=1;i<n;i++) {
        int a,b; scanf("%d%d",&a,&b);
        add(a,b),add(b,a);
    }
    dfs(1,-1);
    for(int i=1;i<=m;i++){
        int a,b; scanf("%d%d",&a,&b);
        node[i]={in[a],out[a],b,i};
    }
    sort(node+1,node+1+m,cmp);
    int l=1,r=0;
    for(int i=1;i<=m;i++){
        while (l<node[i].l) del(dfn[l]),l++;
        while(l>node[i].l) l--,acc(dfn[l]);
        while(r<node[i].r) r++,acc(dfn[r]);
        while (r>node[i].r) del(dfn[r]),r--;
        res[node[i].id]=sum[node[i].k];
    }
    for(int i=1;i<=m;i++) printf("%d\n",res[i]);
}