题目:点击此处
先把树型存储转化成线型存储,然后直接套莫队的板子
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <queue>
#include <stack>
#include <map>
#include <vector>
#include <set>
#define pb push_back
#define ll long long
const int maxn=1e5+5;
using namespace std;
vector<int> v[maxn];
int n,sum=0,m,cnt[maxn],x,y,cnk[maxn],a[maxn],block=sqrt(maxn),start[maxn],End[maxn],arr[maxn];
int ans[maxn];
bool vis[maxn];
struct node
{
int root,num;
int l,r,id;
}pos[maxn];
bool cmp(node a,node b)
{
if(a.l/block!=b.l/block)
return a.l/block<b.l/block;
return a.r<b.r;
}
void dfs(int ra)
{
start[ra]=++sum;
arr[sum]=a[ra];
vis[ra]=1;
for(int i=0;i<v[ra].size();i++)
{
if(!vis[v[ra][i]])
dfs(v[ra][i]);
}
End[ra]=sum;
}
void add(int position) {
cnt[arr[position]]++;
cnk[cnt[arr[position]]]++;
}
void remov1e(int position) {
cnk[cnt[arr[position]]]--;
cnt[arr[position]]--;
}
int main()
{
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++)
scanf("%d",&a[i]);
for(int i=0;i<n-1;i++)
{
scanf("%d%d",&x,&y);
v[x].pb(y);
v[y].pb(x);
}
dfs(1);
for(int i=0;i<m;i++)
{
scanf("%d%d",&pos[i].root,&pos[i].num);
pos[i].l=start[pos[i].root];
pos[i].r=End[pos[i].root];
pos[i].id=i;
}
sort(pos,pos+m,cmp);
int cl=pos[1].l,cr=cl-1;
for(int i=0;i<m;i++)
{
int l=pos[i].l,r=pos[i].r;
// cout<<l<<" "<<r<<endl;
while(cl<l)
{
remov1e(cl);
cl++;
}
while(cl>l)
{
add(cl-1);
cl--;
}
while(cr<r)
{
add(cr+1);
cr++;
}
while(cr>r)
{
remov1e(cr);
cr--;
}
//cout<<cl<<" "<<cr<<" "<<pos[i].num<<endl;
ans[pos[i].id]=cnk[pos[i].num];
}
for(int i=0;i<m;i++)
printf("%d\n",ans[i]);
return 0;
}