Military Problem CodeForces 1006E （dfs序）

J - Military Problem

CodeForces - 1006E

就是一道dfs序的问题

给定一个树， 然后有q次询问。 每次给出u，k， 求以u为根的子树经过深搜的第k个儿子，如果一个节点有多个儿子，按照儿子从小到大的顺序，依次访问，不存在则输出-1。

预处理记录一下每个节点的出入时间 最后每次query直接判断就好了

#include <cstdio>
#include <algorithm>
#include <vector>
using namespace std;

const int maxn = 1e6 + ;
vector<int> G[maxn];
int ans;
int n, q;
int id[maxn], idx[maxn], en[maxn];

void dfs(int u) {
    id[ans] = u;
    idx[u] = ans;
    for (int i = ; i < G[u].size(); i++) {
        int v = G[u][i];
        ans++;
        dfs(v);
    }
    en[u] = ans;
}

int main() {
    scanf("%d%d", &n, &q);
    ans = ;
    for (int i = , temp; i <= n; i++) {
        scanf("%d", &temp);
        G[temp].push_back(i);
    }
    dfs();
    for (int i = , u, k; i < q; i++) {
        scanf("%d%d", &u, &k);
        int temp = idx[u];
        if (id[temp + k -] ==  || idx[id[temp + k - ]] > en[u]) puts("-1");
        else printf("%d\n", id[temp + k - ]);
    }
    return ;
}