题 意:给n个点的一棵树,每个节点,有且只有一个父亲节点。q了询问,输入u,k。问u节点的第k的子节点是什么。如果索引为u的节点没有k个节点则输出-1.
输入范围:
2<=n<=2e5
1<=q<=2e5
样例输入:
9 6
1 1 1 3 5 3 5 7 //第i个节点的父节点是那个 1没有父节点 从第二个开始
3 1
1 5
3 4
7 3
1 8
1 9
样例输出:
3
6
8
-1
9
4
如图:6是3的子节点,而6不是2的子节点。
解题思路: 1.如何确定每个节点的子节点个数,2.如何在o(1)的时间内确定节点u的第k个子节点。发现跑一遍dfs即可,1.记录每个节点子节点的个数 2.记录当前节点是第几个节点 3.记录第几个节点是当前节点。 2,3步的目的呢,是为了完成在o(1)的时间类找到节点为u的第k个节点。因为他是dfs顺所以总体上每个节点的子节点也符合总体的dfs序.
收 获:1.到每一个子节点的最短距离 2.通过2个数组,完成索引和对应的值之间的转化。值在O(1)找到索引,索引在O(1)找到值。
#include<bits/stdc++.h>
using namespace std;
const int maxn = 2e5+5;
typedef long long ll;
vector<int> vec[maxn];
int n,q;
int d[maxn],x[maxn],y[maxn];
int number;
void dfs(int s){
//一个用来记录
d[s] =1;
x[number] = s;
y[s] = number;
number++;
for(int i=0;i<vec[s].size();i++){
int to = vec[s][i];
dfs(to);
d[s] += d[to];
}
}
int main(){
scanf("%d %d",&n,&q);
for(int i=2;i<=n;i++){
int temp;
scanf("%d",&temp);
vec[temp].push_back(i);
}
number = 1;
dfs(1);
for(int i=0;i<q;i++){
int s,t;
scanf("%d %d",&s,&t);
if(d[s] < t) printf("-1\n");
else printf("%d\n",x[y[s]+t-1]);
}
return 0;
}