力扣刷题(day0039)从前序与中序遍历序列构造二叉树

最新推荐文章于 2024-10-11 23:16:03 发布

「已注销」

最新推荐文章于 2024-10-11 23:16:03 发布

阅读量1.3k

点赞数 1

文章标签：数据结构算法 leetcode

本文链接：https://blog.csdn.net/weixin_51600314/article/details/123875589

版权

二叉树前序遍历中序遍历构造递归

关键词由CSDN通过智能技术生成

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

示例 1:

 输入: preorder = [3,9,20,15,7], inorder = [9,3,15,20,7]
 输出: [3,9,20,null,null,15,7]

示例 2:

 输入: preorder = [-1], inorder = [-1]
 输出: [-1]

提示:

1 <= preorder.length <= 3000
inorder.length == preorder.length
-3000 <= preorder[i], inorder[i] <= 3000
preorder 和 inorder 均无重复元素
inorder 均出现在 preorder
preorder 保证为二叉树的前序遍历序列
inorder 保证为二叉树的中序遍历序列

思路：

第一步，如果数组长度为0，则说明是空节点
第二步，如果数组不为空，那么就将前序数组第一个元素作为节点元素
第三步，找到前序数组第一个元素在中序数组中的位置并将其作为切割点
第四步，切割中序数组，切成中序左数组和中序右数组
第五步，切割前序数组，切成前序左数组和后序右数组
第六步，递归处理左区间和右区间

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
private:
    TreeNode* traversal(vector<int>& inorder,int inorderBegin,int inorderEnd,vector<int>& preorder,int preorderBegin,int preorderEnd){
        //第一步，判断数组长度是否为0
        if(preorderEnd==preorderBegin)return NULL;
        //如果数组不为空，将前序数组的第一个元素作为节点
        int rootValue=preorder[preorderBegin];
        TreeNode* root=new TreeNode(rootValue);
        //叶子节点
        if(preorderEnd-preorderBegin==1)return root;
        //找到前序数组第一个元素在中序数组中的位置，将其作为切割点
        int delimiterIndex;
        for(delimiterIndex=inorderBegin;delimiterIndex<inorderEnd;delimiterIndex++){
            if(inorder[delimiterIndex]==rootValue)break;
        }
        //切割中序数组，切割成中序左数组和中序右数组
        int leftInorderBegin=inorderBegin;
        int leftInorderEnd=delimiterIndex;
        int rightInorderBegin=delimiterIndex+1;
        int rightInorderEnd=inorderEnd;
        //切割前序数组，切割成前序左数组和前序右数组
        int leftPreorderBegin=preorderBegin+1;
        int leftPreorderEnd=preorderBegin+1+delimiterIndex-inorderBegin;
        int rightPreorderBegin=preorderBegin+1+delimiterIndex-inorderBegin;
        int rightPreorderEnd=preorderEnd;
        //递归处理左右区间
        root->left=traversal(inorder,leftInorderBegin,leftInorderEnd,preorder,leftPreorderBegin,leftPreorderEnd);
        root->right=traversal(inorder,rightInorderBegin,rightInorderEnd,preorder,rightPreorderBegin,rightPreorderEnd);
        return root;
    }
public:
    TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
        if(preorder.size()==0||inorder.size()==0)return NULL;
        return traversal(inorder,0,inorder.size(),preorder,0,preorder.size());
    }
};