PAT-ADVANCED1138——Postorder Traversal

我的PAT-ADVANCED代码仓：https://github.com/617076674/PAT-ADVANCED

原题链接：https://pintia.cn/problem-sets/994805342720868352/problems/994805345078067200

题目描述：

题目翻译：

1138 后序遍历

假设二叉树中的所有键都是不同的正整数。给定前序和中序遍历序列，你需要输出相应二叉树的后序遍历序列的第一个数字。

输入格式：

每个输入文件包含一个测试用例。对每个测试用例，第一行给出正整数N(<= 50,000)，即二叉树中的节点总数。第二行给出前序序列，第三行给出中序序列。一行中的所有数字都用空格分隔。

输出格式：

对每个测试用例，在一行中打印相应二叉树的后序遍历序列的第一个数字。

输入样例：

7
1 2 3 4 5 6 7
2 3 1 5 4 7 6

输出样例：

3

知识点：递归、二叉树的后序遍历

思路：递归寻找后序遍历的第一个节点

递归函数的定义：void findResult(int preLeft, int preRight, int inLeft, int inRight)

递归终止条件：

当preLeft == preRight时，我们得到后序遍历的第一个节点，即preOrder[preLeft]。

递归过程：

根据前序遍历得到根结点，根据中序遍历区分左右子树，如果左子树不为空，则在左子树中寻找，否则在右子树中寻找。

时间复杂度和时间复杂度均是O(h)，其中h是树的高度。

C++代码：

#include<iostream>
#include<vector>

using namespace std;

vector<int> preOrder;
vector<int> inOrder;
int result;

void findResult(int preLeft, int preRight, int inLeft, int inRight);

int main() {
	int N;
	scanf("%d", &N);
	int num;
	for(int i = 0; i < N; i++) {
		scanf("%d", &num);
		preOrder.push_back(num);
	}
	for(int i = 0; i < N; i++) {
		scanf("%d", &num);
		inOrder.push_back(num);
	}
	findResult(0, preOrder.size() - 1, 0, inOrder.size() - 1);
	printf("%d\n", result);
	return 0;
}

void findResult(int preLeft, int preRight, int inLeft, int inRight) {
	if(preLeft == preRight) {
		result = preOrder[preLeft];
		return;
	}
	int k;
	for(int i = inLeft; i <= inRight; i++) {
		if(inOrder[i] == preOrder[preLeft]) {
			k = i;
			break;
		}
	}
	int numLeft = k - inLeft;
	if(numLeft == 0) {
		findResult(preLeft + numLeft + 1, preRight, k + 1, inRight);
	} else {
		findResult(preLeft + 1, preLeft + numLeft, inLeft, k - 1);
	}
}

C++解题报告：