654. 最大二叉树

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

二叉树的根是数组中的最大元素。
左子树是通过数组中最大值左边部分构造出的最大二叉树。
右子树是通过数组中最大值右边部分构造出的最大二叉树。
通过给定的数组构建最大二叉树，并且输出这个树的根节点。

 

示例 ：

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

      6
    /   \
   3     5
    \    / 
     2  0   
       \
        1
 

提示：

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

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

优化空间很大！但是我今天不想，太困了，要睡了。

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public TreeNode constructMaximumBinaryTree(int[] nums) {
        
        int[] array = getMax(nums);
        TreeNode treeNode = new TreeNode(array[0]);
        if ( nums.length == 1 ) {
        }else {
            if ( array[1] == 0 ) {
                int[] rightArray = new int[nums.length-array[1]-1];
                for ( int i = 0; i < nums.length-array[1]-1; i++ ) {
                    rightArray[i] = nums[i+array[1]+1];
                }
            treeNode.right = constructMaximumBinaryTree(rightArray);
            }else if ( array[1] == nums.length-1) {
                int[] leftArray = new int[array[1]];
                for ( int i = 0; i < array[1]; i++ ) {
                    leftArray[i] = nums[i];
                }
                treeNode.left = constructMaximumBinaryTree(leftArray);
            }else {
                int[] leftArray = new int[array[1]];
                for ( int i = 0; i < array[1]; i++ ) {
                    leftArray[i] = nums[i];
                }
                int[] rightArray = new int[nums.length-array[1]-1];
                for ( int i = 0; i < nums.length-array[1]-1; i++ ) {
                    rightArray[i] = nums[i+array[1]+1];
                }
                treeNode.left = constructMaximumBinaryTree(leftArray);
                treeNode.right = constructMaximumBinaryTree(rightArray);
            }
        } 
        
        return treeNode;
    }

    public int[] getMax(int[] nums) {
        int max = Integer.MIN_VALUE, index = 0;
        for ( int i = 0; i < nums.length; i++ ) {
            if ( nums[i] > max ) {
                max = nums[i];
                index = i;
            }
        }
        int[] ret = {max, index};
        return ret;
    }
}