https://leetcode-cn.com/problems/zhong-jian-er-cha-shu-lcof/
二叉树的问题一般都是用递归。这里给出了前序遍历序列和中序遍历序列,要求重建二叉树。
我们用一个递归函数func来实现这个功能,func函数六个参数:
preorder:前序数组
inorder:中序数组
pre_left:递归时候用于切分前序数组的左边界
pre_right:递归时候用于切分前序数组的右边界
in_left:递归时候用于切分中序数组的左边界
in_right:递归用于切分中序数组的右边界
如下图所示,首先我们先找到树的根节点,即前序序列的第一个元素。然后根据根节点在中序序列中的位置,我们可以划分左子树上的结点和右子树上的结点。pre_left和pre_right括起来的preorder中的元素与in_left和in_right括起来的inorder中的元素是可以一一对应的。
一下代码中用了哈希表来加速查找,因此不适用于结点中有重复值的情况。
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
unordered_map<int, int> index; // 保存中序序列中值与索引的映射关系
TreeNode *func(const vector<int> &preorder, const vector<int> &inorder, int pre_left, int pre_right, int in_left, int in_right){
if(pre_left > pre_right) return nullptr;
int pre_root = pre_left; // 前序遍历第一个结点索引就是根节点索引
int in_root = index[preorder[pre_root]]; // 根据根节点的值定位到其在中序序列中的位置
// 建立根节点
TreeNode *root = new TreeNode(preorder[pre_root]);
int left_size = in_root - in_left; // 根节点索引减去左边索引即根节点对应左子树上结点数目
root->left = func(preorder, inorder, pre_left + 1, pre_left + left_size, in_left, in_root - 1);
root->right = func(preorder, inorder, pre_left + left_size + 1, pre_right, in_root + 1, in_right);
return root;
}
TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
int n = inorder.size();
for(int i = 0; i < n; ++i){
index[inorder[i]] = i;
}
return func(preorder, inorder, 0, n - 1, 0, n - 1);
}
};