Leetcode654. 最大二叉树
题目:
给定一个不含重复元素的整数数组。一个以此数组构建的最大二叉树定义如下:
- 二叉树的根是数组中的最大元素。
- 左子树是通过数组中最大值左边部分构造出的最大二叉树。
- 右子树是通过数组中最大值右边部分构造出的最大二叉树。
通过给定的数组构建最大二叉树,并且输出这个树的根节点。
示例 :
输入:[3,2,1,6,0,5]
输出:返回下面这棵树的根节点:
6
/ \
3 5
\ /
2 0
\
1
题解:递归
java代码:
/**
*
* @param nums
* @return
*/
public static TreeNode constructMaximumBinaryTree(int[] nums) {
if (nums == null || nums.length == 0) return null;
return construct(nums, 0, nums.length);
}
public static TreeNode construct(int[] nums, int l, int r) {
if (l == r) return null;
int max = maxArr(nums, l, r);
TreeNode node = new TreeNode(nums[max]);
node.left = construct(nums, l, max);
node.right = construct(nums, max + 1, r);
return node;
}
/**
* 计算数组最大值所在的索引
*
* @param nums
* @param m
* @param n
* @return
*/
public static int maxArr(int[] nums, int m, int n) {
int max = m;
for (int i = m; i < n; i++) {
if (nums[i] > nums[max]) {
max = i;
}
}
return max;
}