vector(LeetCode重建二叉树)

不秃不秃.

已于 2022-08-22 16:23:18 修改

阅读量1.4k

点赞数

文章标签： leetcode 算法职场和发展

于 2022-08-05 17:58:59 首次发布

本文链接：https://blog.csdn.net/qq_61275084/article/details/126182897

版权

剑指 Offer 07. 重建二叉树

输入某二叉树的前序遍历和中序遍历的结果，请构建该二叉树并返回其根节点。

假设输入的前序遍历和中序遍历的结果中都不含重复的数字。

示例 1:
Input: preorder = [3,9,20,15,7], inorder = [9,3,15,20,7]
Output: [3,9,20,null,null,15,7]
示例 2:
Input: preorder = [-1], inorder = [-1]
Output: [-1]
限制：

0 <= 节点个数 <= 5000

 /**
 * 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) {
        if(preorder.empty()||inorder.empty())return nullptr;
        // 1、建立根节点，然后根据根节点来划分 前序数组 和 中序数组
        TreeNode *root=new TreeNode(*preorder.begin());
        // 2、来找到 中序数组 中根节点的位置
        auto it=find(inorder.begin(),inorder.end(),*preorder.begin());
        // 3、划分中序数组，根据根节点的位置，因为中序数组根节点的左边为左子树，右边为右子树
        vector<int> iLeft(inorder.begin(),it),iRight(it+1,inorder.end());
        // 4、划分前序数组，根据中序数组的左右子树的大小
        int n=iLeft.size();
        vector<int> pLeft(preorder.begin()+1,preorder.begin()+1+n),pRight(preorder.begin()+1+n,preorder.end());
        // 5、递归处理左右子树
        root->left=buildTree(pLeft,iLeft);
        root->right=buildTree(pRight,iRight);
        return root;
    }
};