[剑指-Offer] 7. 重建二叉树(二叉树、递归建树)

1. 题目来源

链接：LeetCode 重建二叉树
来源：LeetCode——《剑指-Offer》专项

2. 题目说明

输入某二叉树的前序遍历和中序遍历的结果，请重建该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。

示例 1：

前序遍历 preorder = [3,9,20,15,7]
中序遍历 inorder = [9,3,15,20,7]

返回如下的二叉树：

 3
/ \
9  20
 /  \
15   7

限制：

• 0 <= 节点个数 <= 5000

3. 题目解析

3.1 方法一：经典方法，递归建树

书中树结构如下图：

二叉树的前序遍历序列的特点： 第一个数字总是树的根结点的值

中序遍历序列的特点： 根结点的值在序列的中间，左子树的结点的值位于根结点的值的左边，而右子树的结点的值位于根结点的值的右边。

因此需要扫描中序遍历序列，才能找到根结点的值。.

如下图所示，前序遍历序列的第一个数字 1 就是根结点的值。扫描中序遍历序列，就能确定根结点的值的位置。根据中序遍历特点，在根结点的值 1 前面的 3 个数字都是左子树结点的值，位于 1 后面的数字都是右子树结点的值。.

由于在中序遍历序列中，有 3 个数字是左子树结点的值，因此左子树总共有 3 个左子结点。同样，在前序遍历的序列中，根结点后面的 3 个数字就是 3 个左子树结点的值，再后面的所有数字都是右子树结点的值。这样就在前序遍历和中序遍历两个序列中，分别找到了左右子树对应的子序列。

既然已经分别找到了左、右子树的前序遍历序列和中序遍历序列，可以用同样的方法分别去构建左右子树。 也就是说，可以用递归的方法去完成。

参见代码如下：

// 执行用时 :20 ms, 在所有 C++ 提交中击败了79.51%的用户
// 内存消耗 :14.5 MB, 在所有 C++ 提交中击败了100.00%的用户

/**
 * 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:
    TreeNode *buildTree(vector<int> &preorder, vector<int> &inorder) {
        return buildTree(preorder, 0, preorder.size() - 1, inorder, 0, inorder.size() - 1);
    }

    TreeNode *buildTree(vector<int> &preorder, int pLeft, int pRight, vector<int> &inorder, int iLeft, int iRight) {
        if (pLeft > pRight || iLeft > iRight) 
            return NULL;
        int i = 0;
        for (i = iLeft; i <= iRight; ++i) {
            if (preorder[pLeft] == inorder[i]) 
                break;
        }
        TreeNode *cur = new TreeNode(preorder[pLeft]);
        cur->left = buildTree(preorder, pLeft + 1, pLeft + i - iLeft, inorder, iLeft, i - 1);
        cur->right = buildTree(preorder, pLeft + i - iLeft + 1, pRight, inorder, i + 1, iRight);
        return cur;
    }
};