codeforces 375D Tree and Queries 树上莫队

https://vjudge.net/problem/CodeForces-375D
题目大意：给一棵有根树，根节点为$1$，每个节点有对应的颜色，$m$个询问$(v,k)$，让你输出在以$v$为根的子树中，出现的次数$>=k$的颜色数目。

思路：子树问题，可以通过$dfs$序将其转换为连续的一段区间，那么就可以上莫队搞了。用$cnt[i]$记录当前区间$[l,r]$颜色为$i$的节点个数，用$cnt[k]$记录$[l,r]$出现的次数$>=k$的颜色数目。

#include<bits/stdc++.h>
#define INF 0x3f3f3f3f
using namespace std;
typedef long long ll;

const int maxn=1e5+5;

int n,m,tot,num;
int a[maxn],pos[maxn],cnt[maxn],cnk[maxn],ans[maxn];
int head[maxn],dfn[maxn],siz[maxn],id[maxn];

struct node
{
    int l,r,k,idx;
    bool operator <(const node &a)const
    {
        if(pos[l]==pos[a.l])
            return r<a.r;
        return pos[l]<pos[a.l];
    }
}q[maxn];

struct Edge
{
    int to,nxt;
}edge[maxn<<1];

inline void addedge(int u,int v)
{
    edge[++tot].to=v,edge[tot].nxt=head[u],head[u]=tot;
}

void dfs(int u,int fa)
{
    dfn[u]=++num,id[num]=u,siz[u]=1;
    int v;
    for(int i=head[u];i;i=edge[i].nxt)
    {
        v=edge[i].to;
        if(v==fa)
            continue;
        dfs(v,u);
        siz[u]+=siz[v];
    }
}

inline void add(int idx)
{
    ++cnt[a[idx]];
    ++cnk[cnt[a[idx]]];
}

inline void sub(int idx)
{
    --cnk[cnt[a[idx]]];
    --cnt[a[idx]];
}

int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
        scanf("%d",&a[i]);
    int u,v;
    for(int i=1;i<n;i++)
    {
        scanf("%d%d",&u,&v);
        addedge(u,v),addedge(v,u);
    }
    int dis=sqrt(n);
    for(int i=1;i<=n;i++)
        pos[i]=(i-1)/dis+1;
    dfs(1,0);
    for(int i=0;i<m;i++)
    {
        scanf("%d%d",&q[i].l,&q[i].k);
        q[i].r=dfn[q[i].l]+siz[q[i].l]-1;
        q[i].l=dfn[q[i].l];
        q[i].idx=i;
    }
    sort(q,q+m);//询问离线
    int l=1,r=0;
    for(int i=0;i<m;i++)
    {
        while(l<q[i].l)
            sub(id[l++]);
        while(l>q[i].l)
            add(id[--l]);
        while(r<q[i].r)
            add(id[++r]);
        while(r>q[i].r)
            sub(id[r--]);
        ans[q[i].idx]=cnk[q[i].k];
    }
    for(int i=0;i<m;i++)
        printf("%d\n",ans[i]);
    return 0;
}