PTA甲级考试真题练习119——1119 Pre- and Post-order Traversals

题目

思路

先序遍历是根左右，后序遍历是左右根
可以注意到如果二叉树某一个结点只有左节点或者只有右结点的时候先序遍历和后序遍历是一样的。
所以在遍历过程中如果发现存在只有左节点或右节点可以判定不止一个中序序列，且我们假定是左节点
那么怎么遍历呢，又注意到，先序遍历开头和后序遍历结尾的结点是一样的，即树的根，那么先序序列的第二个结点是根的一个子节点，我们去后序序列中找到他，发现他会将后序序列分成两个部分，左边部分是根的左节点部分，右边部分是根的右结点部分，此时就可以判断，如果右结点部分为0，则中序序列不唯一。然后递归处理。

代码

#include <iostream>
#include <vector>
using namespace std;
vector<int> pre, post;
vector<int> in;
bool flag = true;
void DFS(int prel,int preh,int postl,int posth) {
	if (prel > preh)  return;
	if (prel == preh) {
		in.emplace_back(pre[prel]);
		return;
	}
	int pias = 0;
	while (post[postl+pias] != pre[prel + 1]) pias++;
	pias++;
	if (pias + prel == preh)  flag = false;
	DFS(prel + 1, prel + pias, postl, postl+pias-1);
	in.emplace_back(pre[prel]);
	DFS(prel + pias + 1, preh, postl + pias, posth - 1);
}

int main()
{
	int n;
	cin >> n;
	pre.resize(n);
	post.resize(n);
	for (int i = 0; i < n; ++i) cin >> pre[i];
	for (int i = 0; i < n; ++i) cin >> post[i];
	DFS(0,n-1,0,n-1);
	printf("%s\n%d", flag ? "Yes" : "No", in[0]);
	for (int i = 1; i < n; ++i) cout << " "<<in[i];
    cout<<endl;
}