Codeforces ~ 1006E ~ Two Strings Swaps(dfs序 + 时间戳)

题意

给你一棵以1为根节点树,然后输入2~n的父节点,Q次查询,每次询问以第u个点为根深搜到的第k个点是哪个,找不到输出-1?

思路

dfs序,记录一下时间戳就好了,即进入每个点和离开每个点的在dfs序中的时间(下标)。

#include<bits/stdc++.h>
using namespace std;
const int MAXN = 2e5 + 5;
int n, Q, in[MAXN], out[MAXN];
vector<int> G[MAXN], v;
void dfs(int u)
{
    in[u] = v.size();
    v.push_back(u);
    for (auto i: G[u]) dfs(i);
    out[u] = v.size()-1;
}
int main()
{
    scanf("%d%d", &n, &Q);
    for (int i = 2; i <= n; i++)
    {
        int u; scanf("%d", &u);
        G[u].push_back(i);
    }
    dfs(1);
    //for (auto i: v) printf("%d %d %d\n", i, in[i], out[i]); puts("");
    while (Q--)
    {
        int u, k; scanf("%d%d", &u, &k);
        if (in[u]+k-1 <= out[u]) printf("%d\n", v[in[u]+k-1]);
        else printf("-1\n");
    }
    return 0;
}
/*
9 6
1 1 1 3 5 3 5 7
3 1
1 5
3 4
7 3
1 8
1 9
*/

猜你喜欢

转载自blog.csdn.net/zscdst/article/details/81078668