[力扣 Hot100]从前序与中序遍历序列构造二叉树

谛听misa

已于 2024-03-27 15:09:11 修改

阅读量525

点赞数 7

CC 4.0 BY-SA版权

分类专栏：力扣Hot100题文章标签： leetcode 算法二叉树

于 2024-03-09 19:48:44 首次发布

本文链接：https://blog.youkuaiyun.com/qq_53627591/article/details/136589209

力扣Hot100题专栏收录该内容

51 篇文章

订阅专栏

本文介绍了一种使用递归方法，通过给定的先序遍历（preorder）和中序遍历（inorder）数组，构建二叉树的算法。通过划分左右子树，从先序遍历找到根节点，然后递归地构造左右子树直到遍历完整个树结构。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目描述

给定两个整数数组 preorder 和 inorder ，其中 preorder 是二叉树的先序遍历， inorder 是同一棵树的中序遍历，请构造二叉树并返回其根节点。
出处

思路

根据先序和中序可以划分左右子树，递归构造子树即可。

代码

class Solution {
public:
    TreeNode* buildSubTree(vector<int>& preorder, vector<int>& inorder, int p_start, int i_start, int i_end) {
        if(p_start>preorder.size()-1||i_start<0||i_start>i_end) return nullptr;
        TreeNode* root=new TreeNode(preorder[p_start]);
        if(i_start==i_end)
            return root;
        int i=i_start;
        while(inorder[i]!=preorder[p_start])i++;
        root->left = buildSubTree(preorder, inorder, p_start+1, i_start, i-1);
        root->right = buildSubTree(preorder, inorder,p_start+(i-i_start+1), i+1, i_end);
        return root;
    }
    TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
        return buildSubTree(preorder, inorder,0,0,inorder.size()-1);
    }
};