剑指offer 07. 重建二叉树

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);

    }
};