【CF106E】Military Problem

题目概述

一棵树，给出q个query每个query(u, k)表示从u开始的第k个dfs的节点，优先dfs编号小的节点。如果不存在的话输出-1。

题目链接

传送门

分析

我们可以用set存边以保持顺序。询问最大的数量级为e5， 如果每次都dfs一下的话很显然会超时。我们选择记录dfs序和它的反函数pre还有子树的大小，和树剖里面的一样。根据子树大小和判断是否存在第k个dfs的点，如果存在根据pre和dfn确定答案。
dfs时间复杂度O(n),每次询问时间复杂度O(1)。

代码

#include <cstdio>
#include <set>

using namespace std;
const int N = 2e5 + 10;
set<int> G[N];
int n, q, cnt;
int dfn[N], pre[N], siz[N];


void dfs(int u){
    siz[u] = 1;
    dfn[u] = ++cnt;
    pre[cnt] = u;
    set<int>::iterator it = G[u].begin();
    for(; it != G[u].end(); ++it){
        dfs(*it);
        siz[u] += siz[*it];
    }
}

int main() {
    scanf("%d%d", &n, &q);
    for (int i = 2, x; i <= n ; ++i) {
        scanf("%d", &x);
        G[x].insert(i);
    }
    dfs(1);
    while (q--){
        int u = 0, k = 0;
        scanf("%d%d", &u, &k);
        printf("%d\n", siz[u] < k ? -1 : pre[dfn[u] + k - 1]);
    }
    return 0;
}