剑指offer07.重建二叉树

意难平丶NJUPT

已于 2022-05-18 20:43:27 修改

阅读量256

点赞数

分类专栏：树递归剑指offer 文章标签： leetcode 算法深度优先

于 2022-05-17 10:58:40 首次发布

本文链接：https://blog.csdn.net/weixin_46589819/article/details/124816567

版权

剑指offer 同时被 3 个专栏收录

11 篇文章 0 订阅

订阅专栏

树

5 篇文章 0 订阅

订阅专栏

递归

1 篇文章 0 订阅

订阅专栏

0517刷题自用

题目描述

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

代码

class Solution {
public:
    TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
        if(preorder.size()==0&&inorder.size()==0) return NULL;
        auto headroot=find(inorder.begin(),inorder.end(),preorder[0]);
        TreeNode*root=new TreeNode(preorder[0]);
        vector<int>inorderleft(inorder.begin(),headroot);
        vector<int>inorderright(headroot+1,inorder.end());
        int n=inorderleft.size();
        vector<int>preorderleft(preorder.begin()+1,preorder.begin()+1+n);
        vector<int>preorderright(preorder.begin()+1+n,preorder.end());
        root->left=buildTree(preorderleft,inorderleft);
        root->right=buildTree(preorderright,inorderright);
        return root;
    }
};