题目:
给定一个不含重复元素的整数数组。一个以此数组构建的最大二叉树定义如下:
二叉树的根是数组中的最大元素。
左子树是通过数组中最大值左边部分构造出的最大二叉树。
右子树是通过数组中最大值右边部分构造出的最大二叉树。
通过给定的数组构建最大二叉树,并且输出这个树的根节点。
示例:
输入:[3,2,1,6,0,5]
输出:返回下面这棵树的根节点:
6
/ \
3 5
\ /
2 0
\
1
源码:
/**
* 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) {
return helper(nums, 0, nums.length - 1);
}
public TreeNode helper(int[] nums, int start, int end) {
if (nums.length == 0) {
return null;
}
if (start > end) {
return null;
}
// 1. 找出 start 到 end 区间最大值元素下标
int bound = findMax(nums, start, end);
// 2. 创建这个下标节点
TreeNode root = new TreeNode(nums[bound]);
// 3. 紧接着递归创建这个节点的左子树和右子树
root.left = helper(nums, start, bound - 1);
root.right = helper(nums, bound + 1, end);
return root;
}
public int findMax(int[] nums,int start, int end) {
int tmp = start;
for (int i = start; i <= end; i++) {
if (nums[i] > nums[tmp]) {
tmp = i;
}
}
return tmp;
}
}