给定一个不含重复元素的整数数组。一个以此数组构建的最大二叉树定义如下:
二叉树的根是数组中的最大元素。 左子树是通过数组中最大值左边部分构造出的最大二叉树。 右子树是通过数组中最大值右边部分构造出的最大二叉树。 通过给定的数组构建最大二叉树,并且输出这个树的根节点。
发现规律:这里明显使用的是前序遍历的框架,先找出根结点,然后是左子树,然后右子树
示例 :
代码语言:javascript复制输入:[3,2,1,6,0,5]
输出:返回下面这棵树的根节点:
6
/
3 5
/
2 0
1
提示: 给定的数组的大小在 [1, 1000] 之间。
解决思路: 1.遍历一次,先求出数组中的最大值index,创建root,值为nums[index] 2.index左边的数组,再次递归进行一次操作,为root的左子树 3.index右边的数组,也是递归,作为root的右子树 4.跳出递归的条件即为start>end
代码语言:javascript复制class Solution {
public static TreeNode constructMaximumBinaryTree(int[] nums) {
if (nums == null || nums.length == 0) {
return null;
}
return constructMaximumBinaryTreeSub(nums, 0, nums.length - 1);
}
public static TreeNode constructMaximumBinaryTreeSub(int[] nums, int start, int end) {
if (start > end) {
return null;
}
int maxIndex = getMaxIndex(start, end, nums);
TreeNode root = new TreeNode(nums[maxIndex]);
root.left = constructMaximumBinaryTreeSub(nums, start, maxIndex - 1);
root.right = constructMaximumBinaryTreeSub(nums, maxIndex 1, end);
return root;
}
public static int getMaxIndex(int start, int end, int[] nums) {
int maxIndex = start;
for (int i = start 1; i <= end; i ) {
int val = nums[i];
int maxVal = nums[maxIndex];
if (val > maxVal) {
maxIndex = i;
}
}
return maxIndex;
}
}