力扣654. 最大二叉树

题目:

给定一个不含重复元素的整数数组 nums 。一个以此数组直接递归构建的 最大二叉树 定义如下:

二叉树的根是数组 nums 中的最大元素。
左子树是通过数组中 最大值左边部分 递归构造出的最大二叉树。
右子树是通过数组中 最大值右边部分 递归构造出的最大二叉树。

示例:

解法

构建二叉树一般用的前序,因为先构建根节点

一般情况来说:如果让空节点(空指针)进入递归,就不加if,如果不让空节点进入递归,就加if限制一下, 终止条件也会相应的调整。

代码

class Solution {
    
    
public:

    TreeNode* traversal(vector<int>& nums, int left, int right) {
    
    
        if (left >= right) return nullptr;

        //分割,找到局部最大值
        int delimitIndex = left;
        for (int i = left + 1; i < right; i++) {
    
    
            if (nums[i] > nums[delimitIndex]) {
    
    
                delimitIndex = i;
            } 
        }
        //创建节点,
        TreeNode* node = new TreeNode(nums[delimitIndex]);
        //左右节点返回值递归
        node->left = traversal(nums, left, delimitIndex);
        node->right = traversal(nums, delimitIndex + 1, right);

        return node;
    }
    TreeNode* constructMaximumBinaryTree(vector<int>& nums) {
    
    
        return traversal(nums, 0, nums.size());
    }
};

猜你喜欢

转载自blog.csdn.net/Matcha_/article/details/114063380
今日推荐