给定一个不含重复元素的整数数组。一个以此数组构建的最大二叉树定义如下:
二叉树的根是数组中的最大元素。
左子树是通过数组中最大值左边部分构造出的最大二叉树。
右子树是通过数组中最大值右边部分构造出的最大二叉树。
通过给定的数组构建最大二叉树,并且输出这个树的根节点。
示例 :
输入:[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;
}
}