【PAT-A1086】Tree Traversals Again（二叉树重建）

【分析】

这道题就是典型的二叉树重建和遍历。从中序遍历的出入栈操作，得到先序序列和中序序列，重建树结构，再得到后序序列。

【代码】

#include <cstdio> 
#include <iostream>
#include <string>
#include <vector>
#include <stack>
using namespace std;
const int maxn = 35;
int n = 0;
vector<int> pre, in, post;
stack<int> st;

struct Node
{
	Node* lchild, *rchild;
	int w;
};

Node* create(int preL, int preR, int inL, int inR) {
	if (preR < preL) {
		return NULL;
	}

	Node* root = new Node;
	root->w = pre[preL];
	// 在中序序列中找到根结点下标
	int k = 0;
	for (int i = inL; i <= inR; i++)
	{
		if (in[i] == pre[preL]) {
			k = i;
			break;
		}
	}
	int Lnum = k - inL, Rnum = inR - k;

	root->lchild = create(preL + 1, preL + Lnum, inL, k - 1);
	root->rchild = create(preL + Lnum + 1, preR, k + 1, inR);
	return root;
}

void postOrder(Node* root) {
	if (root == NULL) {
		return;
	}
	postOrder(root->lchild);
	postOrder(root->rchild);
	post.push_back(root->w);
}

int main() {
	// 输入并获取先序和中序序列
	
	scanf("%d", &n);
	string str;
	int index;
	for (int i = 0; i < 2 * n; i++)
	{
		cin >> str;
		if (str == "Push") {
			cin >> index;
			pre.push_back(index);
			st.push(index);
		}
		else
		{
			int top = st.top();
			st.pop();
			in.push_back(top);
		}
	}

	// 先序后序重建二叉树
	Node* root = create(0, n - 1, 0, n - 1);
	postOrder(root);
	// 输出后序序列
	for (int i = 0; i < n; i++)
	{
		if (i != 0) {
			printf(" ");
		}
		printf("%d", post[i]);
	}

	return 0;
}