Tree and Queries CodeForces - 375D （dfs序+莫队）

题目：点击此处

先把树型存储转化成线型存储，然后直接套莫队的板子

 

#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;
}