1138. Postorder Traversal (25)[树的遍历]

最新推荐文章于 2022-11-25 15:04:59 发布

坚不萌

最新推荐文章于 2022-11-25 15:04:59 发布

阅读量196

点赞数

分类专栏： PAT-甲级文章标签： PAT 甲级递归

本文链接：https://blog.csdn.net/shepherd2010/article/details/79435753

版权

PAT-甲级专栏收录该内容

153 篇文章 3 订阅

订阅专栏

1. 原题：https://www.patest.cn/contests/pat-a-practise/1138

2. 思路：

题意：给出前序和中序，输出后序的第一个元素。
思路：
写个递归，在遍历到第一个元素后，直接返回退出就好了。
已AC。

3. 源码

#include<iostream>
#include<vector>
using namespace std;

int N, flag = 0;
vector<int> pre, ino, re;

void dfs(int pl, int pr, int nl, int nr);
int find_pos(int x, int nl, int nr);

int main()
{
	freopen("in.txt", "r", stdin);
	scanf("%d", &N);
	pre.resize(N);
	ino.resize(N);
	for (int i = 0; i < N; i++)
		scanf("%d", &pre[i]);
	for (int i = 0; i < N; i++)
		scanf("%d", &ino[i]);
	dfs(0, N - 1, 0, N - 1);

	return 0;
}

void dfs(int pl, int pr, int nl, int nr)
{
	if (flag != 0 || pl > pr)	//返回退出
		return;

	int pos = find_pos(pre[pl], nl, nr);
	dfs(pl + 1, pos - nl + pl, nl, pos - 1);
	dfs(pos - nl + pl + 1, pr, pos + 1, nr);
	if (flag == 0)
	{
		printf("%d\n", pre[pl]);
		flag = 1;
	}
	
	return;
}

int find_pos(int x, int nl, int nr)
{
	for (int i = nl; i <= nr; i++)
	{
		if (ino[i] == x)
			return i;
	}

	return 0;
}