题意:
就是有一颗树 然后每次询问 父结点 的 第k个结点是不是他的子嗣。。。是的话就输出这个子嗣。。不是 就输出-1
解析:
突然想到后缀数组的sa 和 x的用法。。就是我们可以用一个id标记当前结点的等级 用idx标记等级ans是哪一个结点。。然后用en标记结点u的子嗣的结束结点
然后每次判断一下就可以了
#include <iostream> #include <cstdio> #include <sstream> #include <cstring> #include <map> #include <cctype> #include <set> #include <vector> #include <stack> #include <queue> #include <algorithm> #include <cmath> #include <bitset> #define rap(i, a, n) for(int i=a; i<=n; i++) #define rep(i, a, n) for(int i=a; i<n; i++) #define lap(i, a, n) for(int i=n; i>=a; i--) #define lep(i, a, n) for(int i=n; i>a; i--) #define rd(a) scanf("%d", &a) #define rlld(a) scanf("%lld", &a) #define rc(a) scanf("%c", &a) #define rs(a) scanf("%s", a) #define pd(a) printf("%d ", a); #define plld(a) printf("%lld ", a); #define pc(a) printf("%c ", a); #define ps(a) printf("%s ", a); #define MOD 2018 #define ULL unsigned long long #define Pair pair<int, int> #define mem(a, b) memset(a, b, sizeof(a)) #define _ ios_base::sync_with_stdio(0),cin.tie(0) //freopen("1.txt", "r", stdin); using namespace std; const int maxn = 1e6+10, INF = 0x7fffffff, LL_INF = 0x7fffffffffffffff; typedef long long LL; int vis[maxn], id[maxn], idx[maxn], en[maxn]; vector<int> G[maxn]; int n, q; int ans; void dfs(int u) { id[ans] = u; idx[u] = ans; // cout<< u << " << " " << idx[u] <<endl; // if(ans == n) return; for(int i=0; i<G[u].size(); i++) { int v = G[u][i]; ans++; dfs(v); } en[u] = ans; } int main() { rd(n), rd(q); int tmp; ans = 1; for(int i=2; i<=n; i++) { rd(tmp); G[tmp].push_back(i); } dfs(1); int u, k; // for(int i=1; i<=n; i++) // cout<< i << " " <<idx[i] << " " << en[i] <<endl; for(int i=0; i<q; i++) { rd(u), rd(k); int temp = idx[u]; if(id[temp+k-1] == 0 || idx[id[temp+k-1]] > en[u]) printf("-1 "); else printf("%d ", id[temp+k-1]); } return 0; }