【LeetCode】654. 最大二叉树(Java)

题目:

Given an integer array with no duplicates. A maximum tree building on this array is defined as follow:

  1. The root is the maximum number in the array.
  2. The left subtree is the maximum tree constructed from left part subarray divided by the maximum number.
  3. The right subtree is the maximum tree constructed from right part subarray divided by the maximum number.

Construct the maximum tree by the given array and output the root node of this tree.

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

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

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

测试样例:

输入: [3,2,1,6,0,5]
输出: 返回下面这棵树的根节点:

      6
    /   \
   3     5
    \    / 
     2  0   
       \
        1

注意:

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

思路

      对给定的数组的指定区段进行一次循环找出该区段的最大值maxNum、记录最大值下标maxIndex并设为根节点的值,再分别对其左右两边进行嵌套处理

Java代码实现:

class TreeNode{
    int val;
    TreeNode left;
    TreeNode right;
    TreeNode(int x){
        val = x;
    }
}
class Solution {
    public TreeNode constructMaximumBinaryTree(int[] nums) {
        return buildTree(nums,0,nums.length-1);
    }
    public TreeNode buildTree(int[] nums,int startIndex,int endIndex){
        if(nums == null || startIndex > endIndex)
            return null;
        
        int maxNum = 0; //找出当前区段最大值所在位置maxIndex
        int maxIndex = startIndex;
        for(int i=startIndex;i<=endIndex;i++){
            if(maxNum < nums[i]){
                maxNum = nums[i];
                maxIndex = i;
            }
        }
                //建立当前根节点并进行嵌套实现
        TreeNode root = new TreeNode(maxNum);
        root.left = buildTree(nums,startIndex,maxIndex-1);
        root.right = buildTree(nums,maxIndex+1,endIndex);
        return root;
    }
}

猜你喜欢

转载自blog.csdn.net/weixin_40849588/article/details/87568609