PAT甲级 1086 Tree Traversals Again (25 分) 前序中序创建二叉树

\quad 从题目给出的入栈和出栈操作，可知入栈顺序是树的前序遍历，本题前序遍历是{1,2,3,4,5,6}；出栈顺序是树的中序遍历，本题的中序遍历是{3,2,4,1,6,5}。题目要求我们求树的后序遍历，因此这道题的关键就是根据前序遍历和中序遍历来创建一棵二叉树。流程如下：

• 1根据前序遍历来确定每次根节点的位置，因为前序遍历先访问的是根节点，所以前序遍历第一个位置就是根节点。
• 2根据根节点和中序遍历将树划分为左右两棵树。
• 3根据第一步和第二步递归的处理左右两棵树。
  \quad 如下图所示，便是这道题的建树过程。
  
  \quad 我用index记录划分边界，例如在最初根节点为1时，index就是在中序遍历中1的位置。根据index将前序遍历和中序遍历划分为两部分。我用pre_l，in_l表示划分后的左子树的前序和中序遍历，pre_r,in_r表示右子树的。这样，我们递归建树，直到前序遍历和中序遍历都为空时结束。利用前序和中序创建二叉树程序如下：

struct TreeNode
{
	int val;
	TreeNode *left, *right;
	TreeNode(int val)
	{
		this->val = val;
		this->left = this->right = NULL;
	}
};

TreeNode* buildTree(vector<int> &pre, vector<int> &in)
{
	TreeNode *root = NULL;
	vector<int> pre_l, pre_r, in_l, in_r;
	int i = 0, index = 0;
	if(pre.empty() && in.empty()) return root; // 没有节点了返回根节点

	root = new TreeNode(pre[0]);
	for(i=0; i<in.size(); i++)
	{
		if(pre[0] == in[i])
		{
			index = i; // 记录根节点在中序遍历中的位置，以便后面的划分
			break;
		}
	}

	// 以index为分界点进行分割
	for(i=0; i<index; i++)
	{
		pre_l.push_back(pre[i+1]);
		in_l.push_back(in[i]);
	}
	for(i=index+1; i<in.size(); i++)
	{
		pre_r.push_back(pre[i]);
		in_r.push_back(in[i]);
	}

    //依次构建分割的部分，缩小区间，直至构建完
	root->left = buildTree(pre_l, in_l);
	root->right = buildTree(pre_r, in_r);

	return root;
}

\quad 拓展：利用中序和后续建二叉树，与根据前序+中序不同的是每次的根节点是后续遍历的最后一个元素。如下图所示：

\quad 程序如下：

struct TreeNode
{
	int val;
	TreeNode *left, *right;
	TreeNode(int val)
	{
		this->val = val;
		this->left = this->right = NULL;
	}
};

TreeNode *buildTree(vector<int> &in, vector<int> &post)
{
	TreeNode *root = NULL;
	int i = 0, index = 0;
	vector<int> in_l, in_r, post_l, post_r;
	if(in.empty() && post.empty()) return root;

	root = new TreeNode(post[post.size()-1]);
	for(i=0; i<in.size(); i++)
	{
		if(in[i]==post[post.size()-1])
		{
			index = i;
			break;
		}
	}

	for(i=0; i<index; i++)
	{
		in_l.push_back(in[i]);
		post_l.push_back(post[i]);
	}
	for(i=index+1; i<in.size(); i++)
	{
		in_r.push_back(in[i]);
		post_r.push_back(post[i-1]);  // 注意这里 
	}

	root->left = buildTree(in_l, post_l);
	root->right = buildTree(in_r, post_r);
	return root;
}

\quad 说了这么多，回归正题，这个题的最终解答是：

#include <bits/stdc++.h>
using namespace std;


struct TreeNode
{
	int val;
	TreeNode *left, *right;
	TreeNode(int val)
	{
		this->val = val;
		this->left = this->right = NULL;
	}
};

TreeNode* buildTree(vector<int> &pre, vector<int> &in)
{
	TreeNode *root = NULL;
	vector<int> pre_l, pre_r, in_l, in_r;
	int i = 0, index = 0;
	if(pre.empty() && in.empty()) return root; // 没有节点了返回根节点

	root = new TreeNode(pre[0]);
	for(i=0; i<in.size(); i++)
	{
		if(pre[0] == in[i])
		{
			index = i; // 记录根节点在中序遍历中的位置，以便后面的划分
			break;
		}
	}

	// 以index为分界点进行分割
	for(i=0; i<index; i++)
	{
		pre_l.push_back(pre[i+1]);
		in_l.push_back(in[i]);
	}
	for(i=index+1; i<in.size(); i++)
	{
		pre_r.push_back(pre[i]);
		in_r.push_back(in[i]);
	}

    //依次构建分割的部分，缩小区间，直至构建完
	root->left = buildTree(pre_l, in_l);
	root->right = buildTree(pre_r, in_r);

	return root;
}

vector<int> post;
void postOrder(TreeNode *root)
{
	if(root!=NULL)
	{
		postOrder(root->left);
		postOrder(root->right);
		post.push_back(root->val);
	}
}

int main(int argc, char const *argv[])
{
	int n;
	cin >> n;
	string s;
	int num;
    stack<int> st;
    vector<int> pre, in;  // 存储前中后序遍历结果
	for (int i = 0; i < 2*n; ++i)
	{
		cin >> s;
		if(s[1]=='o')
		{
			in.push_back(st.top());
			st.pop();
		}
		else
		{
			cin >> num;
			st.push(num);
			pre.push_back(num);
		}
	}
    
    TreeNode *root = buildTree(pre, in);
    postOrder(root);
    cout << post[0];
    for (int i = 1; i < post.size(); ++i)
    {
    	cout << " " << post[i];
    }
	return 0;
}