对于一个二叉树,我们一般有三种不同的遍历方式,分为先序遍历,中序遍历,后序遍历。
先序遍历:根节点 -> 左子树 -> 右子树
中序遍历:左子树 -> 根节点 -> 右子树
后序遍历:左子树 -> 右子树 -> 根节点
根据数组(数组中的值各不相同)构造一个二叉树,我们一般有两种方式,其一是从中序与后序遍历序列构造二叉树(106. 从中序与后序遍历序列构造二叉树),二是从前序与中序遍历序列构造二叉树(105. 从前序与中序遍历序列构造二叉树)。本文以中序与后序为例,我们看如下二叉树和数组。
- 构造一棵二叉树,我们首先要确定的就是节点的值,中序为左根右的方式,而后序为左右根的方式,也就是说,后序数组的最后一个元素即为根元素。
- 根据这样的根元素,我们又可以将中序分为左子树区间,右子树区间,对于后序数组串联在一起的左右区间,我们可以依据中序数组的左子树区间中数的个数将后序数组切分为左区间和右区间。
- 对于切分出来的左区间,后序数组的最后一个元素仍为该左子树的根节点,根据该根节点切分中序数组,得到左左区间,左右区间。
在切分过程中,我们需要接住叶子节点,比如上例,在进行第二次切分时候他的左区间就仅剩下一个,这时候我们可以直接返回,不再对该区间进行操作。 - 重复以上步骤,利用递归的思想,我们最终可以构造出一颗二叉树。
解题代码:
#include <iostream>
#include <string>
#include <algorithm>
#include <vector>
using namespace std;
struct TreeNode {
int val;
TreeNode *left;
TreeNode *right;
};
TreeNode* traversal(vector<int>& inorder, vector<int>& postorder){
if(postorder.size() == 0) return NULL;
TreeNode* root = new TreeNode; // 初始化根节点,并且分配一个内存空间,否则接下来对它进行操作,可能会导致未定义的行为
root->val = postorder[postorder.size() - 1];
if(postorder.size() == 1) return root; // 遍历到了叶子节点,直接返回
// 找到根节点在中序数组的下标,用于切分左右区间
vector<int>::iterator it = find(inorder.begin(), inorder.end(), postorder[postorder.size() - 1]);
int index = distance(inorder.begin(), it);
// 切分左右区间
// 中序
vector<int> left_inorder(inorder.begin(), inorder.begin() + index);
vector<int> right_inorder(inorder.begin() + index + 1, inorder.end());
// 后序
vector<int> left_postorder(postorder.begin(), postorder.begin() + index);
vector<int> right_postorder(postorder.begin() + index, postorder.end() - 1);
// 设置左节点和右节点
root->left = traversal(left_inorder, left_postorder);
root->right = traversal(right_inorder, right_postorder);
return root;
}
int main(){
int arri[] = {9,3,15,20,7};
vector<int> inorder(arri, arri + sizeof(arri) / sizeof(arri[0]));
int arrp[] = {9,15,7,20,3};
vector<int> postorder(arrp, arrp + sizeof(arrp) / sizeof(arrp[0]));
traversal(inorder, postorder);
return 0;
}