CodeForces - 1006E 树的dfs序

本文介绍了一种基于深度优先搜索的算法,用于解决特定类型的树形结构查询问题。具体而言,给定一棵以1为根节点的树及一系列查询,每次查询要求找到以指定节点为根进行深度优先遍历时的第k个访问节点。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

E. Military Problem

time limit per test

3 seconds

memory limit per test

256 megabytes

input

standard input

output

standard output

In this problem you will have to help Berland army with organizing their command delivery system.

There are nn officers in Berland army. The first officer is the commander of the army, and he does not have any superiors. Every other officer has exactly one direct superior. If officer aa is the direct superior of officer bb, then we also can say that officer bb is a direct subordinate of officer aa.

Officer xx is considered to be a subordinate (direct or indirect) of officer yy if one of the following conditions holds:

  • officer yy is the direct superior of officer xx;
  • the direct superior of officer xx is a subordinate of officer yy.

For example, on the picture below the subordinates of the officer 33 are: 5,6,7,8,95,6,7,8,9.

The structure of Berland army is organized in such a way that every officer, except for the commander, is a subordinate of the commander of the army.

Formally, let's represent Berland army as a tree consisting of nn vertices, in which vertex uu corresponds to officer uu. The parent of vertex uucorresponds to the direct superior of officer uu. The root (which has index 11) corresponds to the commander of the army.

Berland War Ministry has ordered you to give answers on qq queries, the ii-th query is given as (ui,ki)(ui,ki), where uiui is some officer, and kiki is a positive integer.

To process the ii-th query imagine how a command from uiui spreads to the subordinates of uiui. Typical DFS (depth first search) algorithm is used here.

Suppose the current officer is aa and he spreads a command. Officer aa chooses bb — one of his direct subordinates (i.e. a child in the tree) who has not received this command yet. If there are many such direct subordinates, then aa chooses the one having minimal index. Officer aa gives a command to officer bb. Afterwards, bb uses exactly the same algorithm to spread the command to its subtree. After bb finishes spreading the command, officer aa chooses the next direct subordinate again (using the same strategy). When officer aa cannot choose any direct subordinate who still hasn't received this command, officer aa finishes spreading the command.

Let's look at the following example:

If officer 11 spreads a command, officers receive it in the following order: [1,2,3,5,6,8,7,9,4][1,2,3,5,6,8,7,9,4].

If officer 33 spreads a command, officers receive it in the following order: [3,5,6,8,7,9][3,5,6,8,7,9].

If officer 77 spreads a command, officers receive it in the following order: [7,9][7,9].

If officer 99 spreads a command, officers receive it in the following order: [9][9].

To answer the ii-th query (ui,ki)(ui,ki), construct a sequence which describes the order in which officers will receive the command if the uiui-th officer spreads it. Return the kiki-th element of the constructed list or -1 if there are fewer than kiki elements in it.

You should process queries independently. A query doesn't affect the following queries.

Input

The first line of the input contains two integers nn and qq (2≤n≤2⋅105,1≤q≤2⋅1052≤n≤2⋅105,1≤q≤2⋅105) — the number of officers in Berland army and the number of queries.

The second line of the input contains n−1n−1 integers p2,p3,…,pnp2,p3,…,pn (1≤pi<i1≤pi<i), where pipi is the index of the direct superior of the officer having the index ii. The commander has index 11 and doesn't have any superiors.

The next qq lines describe the queries. The ii-th query is given as a pair (ui,kiui,ki) (1≤ui,ki≤n1≤ui,ki≤n), where uiui is the index of the officer which starts spreading a command, and kiki is the index of the required officer in the command spreading sequence.

Output

Print qq numbers, where the ii-th number is the officer at the position kiki in the list which describes the order in which officers will receive the command if it starts spreading from officer uiui. Print "-1" if the number of officers which receive the command is less than kiki.

You should process queries independently. They do not affect each other.

Example

input

Copy

9 6
1 1 1 3 5 3 5 7
3 1
1 5
3 4
7 3
1 8
1 9

output

Copy

3
6
8
-1
9
4

【题意】

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

【思路】

    一道dfs序题,需要预处理得到树的dfs序,然后记录一下时间戳就好了

 

参考博客:https://blog.youkuaiyun.com/baidu_35643793/article/details/75433575

 

#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <cmath>
#include <algorithm>
#include <queue>
#include <vector>
#include <map>
#include <stack>
using namespace std;
typedef long long LL;
const int Maxn = 2e5 + 9;
const int Inf = 1 << 30;
const LL INF = 1LL << 60;
int n,q,in[Maxn],out[Maxn],size[Maxn],u,k,cnt;
vector<int> tree[Maxn];

void dfs(int x){
	size[x] = 1;
	in[++cnt] = x;
	out[x] = cnt;
	for (int i = 0; i < tree[x].size(); i++){
		dfs(tree[x][i]);
		size[x] += size[tree[x][i]];
	}
	return;
}

int main ()
{
	while (~scanf("%d%d",&n,&q)){
		for (int i = 2; i <= n; i++){
			scanf("%d",&u);
			tree[u].push_back(i);
		}
		cnt = 0;
		dfs(1);
		for (int i = 1; i <= q; i++){
			scanf("%d%d",&u,&k);
			if (k > size[u]) printf("-1\n");
			else printf("%d\n",in[out[u]+k-1]);
		}
	}
	return 0;
}

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值