题目描述:
给定一个不含重复元素的整数数组 nums 。一个以此数组直接递归构建的 最大二叉树 定义如下:
1.二叉树的根是数组 nums 中的最大元素。
2.左子树是通过数组中 最大值左边部分 递归构造出的最大二叉树。
3.右子树是通过数组中 最大值右边部分 递归构造出的最大二叉树。
返回有给定数组 nums 构建的 最大二叉树 。
示例:输入nums = [3,2,1,6,0,5] 返回如下图所示的树
算法思想:
题目已经明确给出了思路–运用递归,对于构造二叉树的问题,根节点要做的就是把想办法把自己构造出来。我们先找到数组中的最大值,把root做出来,然后对左右数组递归调用,作为root的左右子树。
这里涉及到左右两个区间,为了把两个区间独立出来,故使用build函数记录low和high的位置,代码如下所示:
public TreeNode constructMaximumBinaryTree(int[] nums) {
return build(nums, 0, nums.length-1);
}
public TreeNode build(int[] nums, int low, int high) {
//递归终止
if(low > high) return null;
//找出最大元素的下标
int max = low;
for(int i = low + 1; i <= high; ++i) {
if(nums[i] > nums[max]) {
max = i;
}
}
TreeNode root = new TreeNode(nums[max]);
//递归左端
root.left = build(nums, low, max - 1);
//递归右端
root.right = build(nums, max + 1, high);
return root;
}