剑指 Offer 07. 重建二叉树（分治）

系列文章目录

---

---

题目

一、分析

根据前序遍历，根->左->右 的遍历顺序，整棵树的根节点就是前序遍历的首元素，我们可以很快找到左子树的根节点，假设当前的根节点为i，那么它的左子树根节点就是i+1；
根据中序遍历，左->根->右 的遍历顺序，可以划分左子树和右子树的位置，右子树的区间的前序遍历第一个就是右子树根节点。
最后通过递归，每次循环都能找到左子树和右子树的根节点。

二、代码

/**
 * 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++)
            dic[inorder[i]] = i;
        return recur(0, 0, inorder.size() - 1);
    }
private:
    vector<int> preorder;
    unordered_map<int, int> dic;
    TreeNode* recur(int root, int left, int right) {
    
     
        if(left > right) return nullptr;                        // 递归终止
        TreeNode* node = new TreeNode(preorder[root]);          // 建立根节点
        int i = dic[preorder[root]];                            // 划分根节点、左子树、右子树
        node->left = recur(root + 1, left, i - 1);              // 开启左子树递归
        node->right = recur(root + i - left + 1, i + 1, right); // 开启右子树递归
        return node;                                            // 回溯返回根节点
    }
};

总结

今天复习了树的知识，练！！！