leetcode刷题：二叉树18（最大二叉树）

654.最大二叉树

力扣题目地址

给定一个不含重复元素的整数数组。一个以此数组构建的最大二叉树定义如下：

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

通过给定的数组构建最大二叉树，并且输出这个树的根节点。

示例 ：

提示：

给定的数组的大小在 [1, 1000] 之间。

遍历，找最大值和其位置。数组分割为左右子树数组。创建结点添加左右子树。递归终止条件为左右数组都为空，即为叶子结点。

package com.programmercarl.tree;

/** * @ClassName ConstructMaximumBinaryTree * @Descriotion TODO * @Author nitaotao * @Date 2022/7/5 15:45 * @Version 1.0 * 654. 最大二叉树 * https://leetcode.cn/problems/maximum-binary-tree/ **/
public class ConstructMaximumBinaryTree {
    
    public TreeNode constructMaximumBinaryTree(int[] nums) {
    
        if (nums.length == 0) {
    
            return null;
        }
        int max = -1;
        int maxIndex = 0;
        for (int i = 0; i < nums.length; i++) {
    
            if (nums[i] > max) {
    
                max = nums[i];
                maxIndex = i;
            }
        }
        //找到当前最大值
        TreeNode root = new TreeNode(nums[maxIndex]);
        // 6 - 3 = 3
        int[] leftNums = new int[maxIndex];
        // 6 - 3 - 1 = 2
        int[] rightNums = new int[nums.length - maxIndex - 1];
        for (int i = 0; i < nums.length; i++) {
    
            if (i < maxIndex) {
    
                leftNums[i] = nums[i];
            } else if (i > maxIndex) {
    
                rightNums[i - maxIndex - 1] = nums[i];
            }
        }
// System.out.println(leftNums);
// System.out.println(rightNums);
        root.left = constructMaximumBinaryTree(leftNums);
        root.right = constructMaximumBinaryTree(rightNums);
        return root;
    }

    public static void main(String[] args) {
    
        new ConstructMaximumBinaryTree().constructMaximumBinaryTree(new int[]{
    3,2,1,6,0,5});
    }
}