CF375D Tree and Queries

题目

luogu

思路

唯有水题暖人心
只用子树的树，当然dfs序列
区间查询出现k次的数字的个数
nub[i]记录出现次数大于i的数字个数
发现只是O(1)的转移，想想就OK了

代码

ps:为了更好地阅读，我加一段cpp吧

void add(int x)
{   
    hav[x]++;
    nub[hav[x]]++;
}
void delet(int x)
{
    nub[hav[x]]--;
    hav[x]--;
}
FOR(i,1,m)
{
    while(l > Q[i].s) add(a[b[--l]]);
    while(l < Q[i].s) delet(a[b[l++]]);
    while(r < Q[i].t) add(a[b[++r]]);
    while(r > Q[i].t) delet(a[b[r--]]);
    Q[i].ans=nub[Q[i].k];
}

#include <bits/stdc++.h>
#define FOR(i,a,b) for(int i=a;i<=b;++i)
using namespace std;
const int maxn=1e5+7;
int n,m,a[maxn],b[maxn],begg[maxn],endd[maxn],js,nub[maxn],hav[maxn];;
int head[maxn<<1],tot,belong[maxn];
struct node
{
    int s,t,k,id,ans;
}Q[maxn];
bool cmp1(const node &a,const node &b){ return belong[a.s]==belong[b.s] ? a.t<b.t : a.s<b.s;}
bool cmp2(const node &a,const node &b){ return a.id<b.id;}
struct edge
{
    int v,nxt;
}e[maxn<<1];
void add_edge(int u,int v)
{
    e[++tot].v=v;
    e[tot].nxt=head[u];
    head[u]=tot;
}
int read()
{
    int x=0,f=1;char s=getchar();
    for(;s<'0'||s>'9';s=getchar()) if(s=='-') f=-1;
    for(;s>='0'&&s<='9';s=getchar()) x=x*10+s-'0';
    return x*f;
}
void dfs(int u,int f)
{
    b[++js]=u;
    begg[u]=js;
    for(int i=head[u];i;i=e[i].nxt)
    {
        int v=e[i].v;
        if(v==f) continue;
        dfs(v,u);
    }
    endd[u]=js;
}
int main()
{
    n=read(),m=read();
    int k=sqrt(n);
    FOR(i,1,n) a[i]=read();
    FOR(i,1,n) belong[i]=(i-1)/k+1;
    FOR(i,1,n-1)
    {
        int x=read(),y=read();
        add_edge(x,y);
        add_edge(y,x);
    }
    dfs(1,0);
    FOR(i,1,m)
    {
        int x=read(),y=read();
        Q[i].s=begg[x];
        Q[i].t=endd[x];
        Q[i].k=y;
        Q[i].id=i;
    }
    sort(Q+1,Q+1+m,cmp1);
    int l=1,r=0;
    FOR(i,1,m)
    {
        while(l > Q[i].s) hav[a[b[--l]]]++,nub[hav[a[b[l]]]]++;
        while(l < Q[i].s) nub[hav[a[b[l]]]]--,hav[a[b[l++]]]--;
        while(r < Q[i].t) hav[a[b[++r]]]++,nub[hav[a[b[r]]]]++;
        while(r > Q[i].t) nub[hav[a[b[r]]]]--,hav[a[b[r--]]]--;
        Q[i].ans=nub[Q[i].k];
    }
    sort(Q+1,Q+1+m,cmp2);
    FOR(i,1,m) printf("%d\n",Q[i].ans);
}