题目描述:
给定一个不含重复元素的整数数组 nums 。一个以此数组直接递归构建的 最大二叉树 定义如下:
- 二叉树的根是数组 nums 中的最大元素。
- 左子树是通过数组中 最大值左边部分 递归构造出的最大二叉树。
- 右子树是通过数组中 最大值右边部分 递归构造出的最大二叉树。
- 返回有给定数组 nums 构建的 最大二叉树 。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/maximum-binary-tree
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解题思路:
- 题目要求将数组中的最大值作为根节点,最大值左边的值作为左子树,最大值右边的数作为右子树,这题有点像二分查找,找到最大值将数组一分为二,然后继续找分开的两个数组中最大值,所以我们可以定义一个函数,用于专门找数组的最大值。然后定义一个构建二叉树的函数来调用求数组最大值。
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode() {}
* TreeNode(int val) { this.val = val; }
* TreeNode(int val, TreeNode left, TreeNode right) {
* this.val = val;
* this.left = left;
* this.right = right;
* }
* }
*/
class Solution {
public TreeNode constructMaximumBinaryTree(int[] nums) {
return construct(nums, 0, nums.length);
}
//递归构建二叉树
public TreeNode construct(int[] nums, int left, int right) {
if (left == right) return null;
int maxIndex = max(nums, left, right);
TreeNode root = new TreeNode(nums[maxIndex]);
root.left = construct(nums, left, maxIndex);
root.right = construct(nums, maxIndex + 1, right);
return root;
}
//找到数组中最大值的下标
public int max(int[] nums, int left, int right) {
int maxIndex = left;
for (int i = left; i < right; i++) {
if (nums[i] > nums[maxIndex]) {
maxIndex = i;
}
}
return maxIndex;
}
}