CodeForces - 1006E Military Problem （ DFS

最新推荐文章于 2020-07-11 15:19:16 发布

原创最新推荐文章于 2020-07-11 15:19:16 发布 · 416 阅读

0 ·

CC 4.0 BY-SA版权

online judge CodeForces 同时被 2 个专栏收录

82 篇文章

订阅专栏

搜索 DFS/BFS

47 篇文章

订阅专栏

本文介绍了一种使用深度优先搜索（DFS）遍历树形结构的方法，并实现了基于节点的查询功能。通过预处理得到每个节点的进入时间和退出时间，可以快速查找指定节点的第k个子节点或返回-1表示不存在。

题意：

#include <algorithm>
#include <cstdio>
#include <cstring>
#include <iostream>
#include <bits/stdc++.h>
using namespace std;

#define pb push_back
#define ll long long
const int MAXN = 2e5+10;

int in[MAXN], out[MAXN];
int ans[MAXN];
vector<int> G[MAXN];
int tot;
void dfs(int u) {
    in[u] = ++tot;
    ans[tot] = u;
    for(int i = 0; i < G[u].size(); ++i) {
        int v = G[u][i];
        dfs(v);
    }
    out[u] = tot;
}
int main() {
    #ifndef ONLINE_JUDGE
        freopen("in.txt","r",stdin);
    #endif // ONLINE_JUDGE
    ios::sync_with_stdio(false);
    tot = 0;
    int n, q, x;
    cin >> n >> q;
    for(int i = 2; i <= n; i++) {
        cin >> x;
        G[x].pb(i);
    }
    dfs(1);
    while(q--) {
        int u, k;
        cin >> u >> k;
        if(out[u]-in[u]+1 < k)
            cout << "-1\n";
        else
            cout << ans[in[u]+k-1] << endl;
    }
    return 0;
}