剑指 Offer 07. 重建二叉树

原题：剑指 Offer 07. 重建二叉树

二叉树的前序遍历：根结点-左子树-右子树
二叉树的中序遍历：左子树-根结点-右子树

根据二叉树的这两个遍历性质，可以给出这道题的思路：

1. 前序遍历首元素为根结点root的值
2. 在中序遍历中找到根结点root所在的位置i，可以将中序遍历分为 左子树-根结点-右子树
3. 根据中序遍历中左右子树的结点数目，可以将前序遍历分为 根结点-左子树-右子树。问题就回到了第1步。

需要注意的是：

1. 将使用哈希表dir记录每个结点在中序遍历中的位置，后续根据结点值就能在O(1)时间内找到中序遍历下标
2. 递归终止的条件，到达空节点时：left > right
3. 左子树的函数形参问题
   左子树根结点位置在当前root结点右侧一位，即root+1
   左子树在中序遍历中的范围为[left,i-1]
4. 右子树的函数形参问题
   右子树根结点位置在前序遍历中左子树结束后首位，即root+i-left+1（根结点位置+左子树大小+1），这里左子树的大小又是根据中序遍历当中的索引位置来计算的。
   右子树在中序遍历中的范围为[i+1,right]

/**
 * 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) {
        this->preorder=preorder;
        for(int i=0;i<inorder.size();i++){
            dir[inorder[i]]=i;
        }
        TreeNode* p=recurTree(0,0,inorder.size()-1);
        return p;
    }
private:
    unordered_map<int,int> dir;
    vector<int> preorder;
    TreeNode* recurTree(int root,int left,int right){
        if(left > right) return nullptr;
        TreeNode* node=new TreeNode(preorder[root]);
        int i=dir[preorder[root]];
        node->left = recurTree(root+1,left,i-1);
        node->right = recurTree(root+i-left+1,i+1,right);
        return node;
    }
};

复杂度分析：

• 时间复杂度O(N)：N为结点的数量。在建立哈希表的过程中需遍历inorder，占用O(N)。递归建立N个结点，每层递归中的节点建立、搜索操作占用O(1)，因此时间复杂度为O(N)。
• 空间复杂度O(N)： HashMap 使用 O(N) 额外空间。最差情况下，树退化为链表，递归深度达到 N ，占用 O(N) 额外空间；最好情况下，树为满二叉树，递归深度为 log⁡N，占用 O(log⁡N) 额外空间。