ACwing刷题笔记 18.重建二叉树

题目

思路

思路：只要我们在中序遍历中定位到根节点，那我们就可以直到左子树和右子树中节点的数目，那么我们也就知道左子树和右子树的遍历结果。然后就可以递归构造左子树和右子树。

 在中序遍历对根节点进行定位时，一种简单的方法是直接扫描中序遍历的结果找到根节点，但是时间复杂度高，我们采用哈希表来快速定位根节点（空间换时间），对于哈希映射中每一个键值对，键表示一个元素，值表示其在中序遍历中的出现位置。

代码

/**
 * 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 {
private:
    unordered_map<int,int> index;// 创建哈希表 快速定位中序遍历根节点
    
public:
    TreeNode* myBuildTree(vector<int>& preorder,vector<int>& inorder,int pLeft,int pRight,int iLeft,int iRight)
    {
        if(pLeft > pRight)
        {
            // 递归出口
            return nullptr;
        }
        
        // 去除先序遍历的第一个节点 就是二叉树的根节点
        int pre_root = pLeft;
        // 定位到中序遍历的根节点
        int in_root = index[preorder[pre_root]];// 取出根节点 定位哈希表中的值
        
        // 建立根节点
        TreeNode* root = new TreeNode(preorder[pre_root]);
        
        // 计算左子树的节点数目
        int left_num = in_root - iLeft;
        
        // 递归构造左子树
        root->left = myBuildTree(preorder,inorder,pLeft + 1,pLeft + left_num,iLeft,in_root - 1);
        
        // 递归构造右子树
        root->right = myBuildTree(preorder,inorder,pLeft + left_num + 1,pRight,in_root + 1,iRight);
        
        return root;
        
    }
    TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
        int n = preorder.size();
        // 构造哈希映射 快速定位中序遍历根节点
        for(int i = 0; i < n; i++)
        {
            index[inorder[i]] = i;
        }
        
        return myBuildTree(preorder,inorder,0,n - 1,0,n - 1);
    }
};