给定一个不含重复元素的整数数组。一个以此数组构建的最大二叉树定义如下:
二叉树的根是数组中的最大元素。
左子树是通过数组中最大值左边部分构造出的最大二叉树。
右子树是通过数组中最大值右边部分构造出的最大二叉树。
通过给定的数组构建最大二叉树,并且输出这个树的根节点。
这道题其实有点像有先序遍历和中序遍历构造二叉树,给了树根节点的值和左右子树的构造方式,以此来构造二叉树。
public static TreeNode constructMaximumBinaryTree(int[] nums) {
return recursion(nums, 0, nums.length - 1);
}
public static TreeNode recursion(int[] nums, int start, int end) {
//结束递归条件
if (start > end) {
return null;
}
//这个判断也可以不要
if (start == end) {
return new TreeNode(nums[start]);
}
int max = nums[start];
int k = start;
for (int i = start; i <= end; i++) {
if (max <= nums[i]) {
k = i;
max = nums[i];
}
}
TreeNode root = new TreeNode(max);
root.left = recursion(nums, start, k - 1);
root.right = recursion(nums, k + 1, end);
return root;
}