【剑指offer】07. 重建二叉树

最新推荐文章于 2024-08-20 17:09:37 发布

唯有拼搏

最新推荐文章于 2024-08-20 17:09:37 发布

阅读量157

点赞数 1

分类专栏： leetcode刷题文章标签：算法 c++ 数据结构

本文链接：https://blog.csdn.net/ajin156/article/details/126397470

版权

leetcode刷题专栏收录该内容

2 篇文章 0 订阅

订阅专栏

这篇博客讨论了如何利用前序遍历和中序遍历的结果来递归地构建二叉树。通过分析前序遍历确定根节点，然后在中序遍历中找到根节点位置，以此划分左右子树并进行递归构建。具体实现中，定义了一个名为`build`的辅助函数，根据给定的两个遍历序列的边界进行操作。

摘要由CSDN通过智能技术生成

1.题目

2.解题思路

用递归的思想解决问题：

首先，前序遍历是“根左右”，而中序遍历是“左根右”，所以由前序遍历能确定根节点，然后在中序遍历中找到该节点的位置，由该节点可划分左右子树，我们可以发现，数字1是根节点，并将二叉树分成了(247)和(3568)两棵子树。并且可以判断出左右字数的长度，由此构建左右子树；

假设，preStart为前序遍历的起始位置，int num = preorder[preStart]为根节点的值，所以就可以构建TreeNode* root = new TreeNode(num)的一个新的二叉树，再找出根节点在中序遍历中的位置，即index，int leftSize = index - inStart为左子树的长度，然后构建二叉树的左右子树，不断递归；

class Solution {
public:
    TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) 
    {

        return build(preorder, inorder, 0, preorder.size() - 1, 0, inorder.size() - 1);
    }

    TreeNode* build(vector<int>& preorder, vector<int>& inorder, int preStart , int preEnd,    int inStart, int inEnd)
    {

        if(preStart > preEnd) 
            return nullptr;
        int num = preorder[preStart];
        TreeNode* root = new TreeNode(num);
        int index = 0;
        for(int i = inStart; i <= inEnd; i++)
        {
            if(inorder[i] == num)
            {
                index = i;
                break;
            }
        }
        int leftSize = index - inStart;
        
        root->left = build(preorder, inorder, preStart + 1, preStart + leftSize, inStart, index - 1);
        root->right = build(preorder, inorder, preStart + leftSize + 1, preEnd, index + 1, inEnd);

        return root;
    }

};