LeetCode 654.最大二叉树

时间:2023-04-07 19:59:01

1.题目:

给定一个不重复的整数数组 nums 。 最大二叉树 可以用下面的算法从 nums 递归地构建:

创建一个根节点,其值为 nums 中的最大值。 递归地在最大值 左边 的 子数组前缀上 构建左子树。 递归地在最大值 右边 的 子数组后缀上 构建右子树。 返回 nums 构建的 最大二叉树 。


示例 1:

输入:nums = [3,2,1,6,0,5] 输出:[6,3,5,null,2,0,null,null,1] 解释:递归调用如下所示:

  • [3,2,1,6,0,5] 中的最大值是 6 ,左边部分是 [3,2,1] ,右边部分是 [0,5] 。
  • [3,2,1] 中的最大值是 3 ,左边部分是 [] ,右边部分是 [2,1] 。
  • 空数组,无子节点。
  • [2,1] 中的最大值是 2 ,左边部分是 [] ,右边部分是 [1] 。
  • 空数组,无子节点。
  • 只有一个元素,所以子节点是一个值为 1 的节点。
  • [0,5] 中的最大值是 5 ,左边部分是 [0] ,右边部分是 [] 。
  • 只有一个元素,所以子节点是一个值为 0 的节点。
  • 空数组,无子节点。

来源:力扣(LeetCode) 链接:https://leetcode.cn/problems/maximum-binary-tree 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。


2.代码:

递归法:

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    public TreeNode constructMaximumBinaryTree(int[] nums) {
        return findTreeNode(nums,0,nums.length);


    }
    public TreeNode findTreeNode(int[] nums,int leftIndex,int rightIndex){
        if(rightIndex-leftIndex<1){//没有元素了
            return null;
        }
        if(rightIndex-leftIndex==1){//只有一个元素
            return new TreeNode(nums[leftIndex]);//直接返回这一个元素构成的二叉树
        }
        int maxIndex=leftIndex;//注意这里的赋值不能是0、nums.length等等,直接用传进来的参数就好
         int maxVal = nums[maxIndex];
        for(int i=leftIndex; i<rightIndex; i++){//注意这里是不断需要判断最大值的,不是只确定根节点是最大值就行
            if(nums[i]>maxVal){
                maxVal=nums[i];
                maxIndex=i;
            }

        }
        TreeNode root = new TreeNode(maxVal);
        root.left=findTreeNode(nums,leftIndex,maxIndex);//坚持前闭后开的原则
        root.right=findTreeNode(nums,maxIndex+1,rightIndex);
        return root;

    }
}