问题描述
给定一个不含重复元素的整数数组。一个以此数组构建的最大二叉树定义如下:
二叉树的根是数组中的最大元素。
左子树是通过数组中最大值左边部分构造出的最大二叉树。
右子树是通过数组中最大值右边部分构造出的最大二叉树。
通过给定的数组构建最大二叉树,并且输出这个树的根节点。
问题分析
分析:根据题目的定义,一个最大二叉树的子树也是最大二叉树,所以最大二叉树是按照递归的方式来生成的,这也是树最常见的定义方式。
我们首先看第一次递归的过程,不难发现,我们只要得到6的下标(最大值),将其作为根节点,其左右孩子是其左边区间和右边区间中最大的值,递归地得到这两个节点后再重复这个过程就可以得到整个树。
注意一下边界条件,我们传递的应该是数组、左边界和右边界(因为我们需要对数组进行划分):
- 当左边界大于右边界时,返回null;
- 当左边界等于右边界时,此时是叶子节点,返回该节点;
Code
class Solution {
public TreeNode constructMaximumBinaryTree(int[] nums) {
return func(nums, 0, nums.length-1);
}
public TreeNode func(int[] nums, int l, int r) {
// 边界条件
if (l > r) return null;
if (l == r) return new TreeNode(nums[l]);
// 得到下标
int idx = getGreatestNum(nums, l, r);
// 递归过程,划分区间
TreeNode root = new TreeNode(nums[idx]);
root.left = func(nums, l, idx-1);
root.right = func(nums,idx+1,r);
return root;
}
public int getGreatestNum(int[] nums, int l, int r) {
int max = l--;
while (++l <= r)
max = nums[max] > nums[l] ? max : l;
return max;
}
}
结果: