PAT-ADVANCED1086/Data Structures and Algorithms7-5——Tree Traversals Again

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

我的Data Structures and Algorithms代码仓：https://github.com/617076674/Data-Structures-and-Algorithms

原题链接：

PAT-ADVANCED1086：https://pintia.cn/problem-sets/994805342720868352/problems/994805380754817024

Data Structures and Algorithms7-5：https://pintia.cn/problem-sets/16/problems/667

题目描述：

PAT-ADVANCED1086、Data Structures and Algorithms7-5：

题目翻译：

1086 树的再次遍历

可以使用堆栈以非递归方式实现二叉树的中序遍历。 例如，当遍历6节点二叉树（编号为1到6的键）时，堆栈操作为：push(1); push(2); push(3); pop(); pop(); push(4); pop(); pop(); push(5); push(6); pop(); pop()。 然后可以从该操作序列生成唯一的二叉树（如下图所示）。 你的任务是给出这棵树的后序遍历序列。

输入格式：

每个输入文件包含一个测试用例。在每个测试用例中，第一行给出一个正整数N(<= 30)，代表树的节点个数（节点编号为1 ~ N）。接下来的2N行，每行以下述形式描述了一次栈操作："Push X"，这里X指进栈的节点编号；或"Pop"，弹出栈顶元素。

输出格式：

对每个测试用例，在一行中打印后序遍历的结果。题目保证一定有且只有一个解。每个数字间隔一个空格，行末不得有多余空格。

输入样例：

6
Push 1
Push 2
Push 3
Pop
Pop
Push 4
Pop
Pop
Push 5
Push 6
Pop
Pop

输出样例：

3 4 2 6 5 1

知识点：二叉树前序遍历和中序遍历的非递归实现、二叉树的后序遍历

思路：用前序遍历和中序遍历重建二叉树，再后序遍历得到结果

前序遍历的本质是：在第一次访问该节点时就输出该节点信息。

中序遍历的本质是：在第二次访问该节点时就输出该节点信息。

对应本题，节点的入栈顺序就是其前序遍历结果，节点的出栈顺序就是其中序遍历结果。而任给一棵二叉树前序遍历和中序遍历信息，我们一定能唯一地重构出这棵二叉树。

时间复杂度和空间复杂度均是O(N)。

C++代码：

#include<iostream>
#include<string>
#include<vector>
#include<stack>

using namespace std;

struct node {
	int lchild;
	int rchild;
};

int N;
node Node[31];
int index = 1;
vector<int> preOrder;
vector<int> inOrder;
vector<int> postOrder;

int create(int leftPre, int rightPre, int leftIn, int rightIn);
void postOrderTraversal(int root); 

int main(){
	cin >> N;
	string operate;
	int num;
	stack<int> Stack;
	for(int i = 0; i < 2 * N; i++){
		cin >> operate;
		if(operate.compare("Push") == 0){
			cin >> num;
			preOrder.push_back(num);
			Stack.push(num);
		}else if(operate.compare("Pop") == 0){
			inOrder.push_back(Stack.top());
			Stack.pop();
		}
	}
	int root = create(0, preOrder.size() - 1, 0, inOrder.size() - 1);
	postOrderTraversal(root);
	for(int i = 0; i < postOrder.size(); i++){
		cout << postOrder[i];
		if(i != postOrder.size() - 1){
			cout << " ";
		}
	}
	cout << endl;
	return 0;
}

int create(int leftPre, int rightPre, int leftIn, int rightIn){
	if(leftPre > rightPre){
		return -1;
	}
	int root = preOrder[leftPre];
	int k;
	for(k = leftIn; k <= rightIn; k++){
		if(inOrder[k] == preOrder[leftPre]){
			break;
		}
	}
	int numLeft = k - leftIn;
	Node[root].lchild = create(leftPre + 1, leftPre + numLeft, leftIn, k - 1);
	Node[root].rchild = create(leftPre + numLeft + 1, rightPre, k + 1, rightIn);
	return root;
}

void postOrderTraversal(int root){
	if(root == -1){
		return;
	}
	postOrderTraversal(Node[root].lchild);
	postOrderTraversal(Node[root].rchild);
	postOrder.push_back(root);
}

C++解题报告：