题目：点击此处
先把树型存储转化成线型存储，然后直接套莫队的板子
#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; }
