题意:给定一棵树,每个点有颜色,m个询问vi,ki,问以vi为根的子树中有多少种颜色出现的次数>=ki。首先我们可以通过dfn把子树询问变成序列询问。(不会看代码,一颗子树中的顶点的dfn一定是连续的,画画就知道了)然后就是经典的莫队了,至于ans怎么更新,见代码啦~
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
using namespace std;
#define N 100005
int n,m,block,aa[N],a[N],f[N],h[N],num=0,ANS[N],ans[N];
int in[N],out[N],dfn=0,fa[N];
bool vis[N];
struct query{
int l,r,k,id,block;
}q[N];
struct edge{
int to,next;
}data[N];
inline int read(){
int x=0;char ch=getchar();
while(ch<'0'||ch>'9') ch=getchar();
while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
return x;
}
void dfs(int x){
in[x]=++dfn;
for(int i=h[x];i;i=data[i].next){
int y=data[i].to;
if(fa[x]==y) continue;
fa[y]=x;dfs(y);
}
out[x]=dfn;
}
inline bool cmp(query x,query y){
return x.block==y.block?x.r<y.r:x.block<y.block;
}
inline void update(int x){
//cnt->cnt-1 只会影响ans[cnt];cnt->cnt+1 只会影响ans[cnt+1]
if(vis[x]) ans[f[a[x]]--]--;
else ans[++f[a[x]]]++;
vis[x]^=1;
}
int main(){
// freopen("a.in","r",stdin);
n=read();m=read();block=sqrt(n);
for(int i=1;i<=n;++i) aa[i]=read();
for(int i=1;i<n;++i){
int x=read();int y=read();
data[++num].to=y;data[num].next=h[x];h[x]=num;
data[++num].to=x;data[num].next=h[y];h[y]=num;
}
dfs(1);
//把颜色映射到dfn序上去,转化为序列上莫队
for(int i=1;i<=n;++i) a[in[i]]=aa[i];
for(int i=1;i<=m;++i){
int x=read();q[i].l=in[x];q[i].r=out[x];
q[i].k=read();q[i].id=i;
q[i].block=(q[i].l-1)/block;
}
sort(q+1,q+m+1,cmp);
int l=1,r=0;
for(int i=1;i<=m;++i){
for(;l<q[i].l;++l) update(l);
for(;l>q[i].l;--l) update(l-1);
for(;r<q[i].r;++r) update(r+1);
for(;r>q[i].r;--r) update(r);
ANS[q[i].id]=ans[q[i].k];
}
for(int i=1;i<=m;++i) printf("%d\n",ANS[i]);
return 0;
}