【LeetCode】654. 最大二叉树

题目描述:

给定一个不含重复元素的整数数组 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;
    }
}

猜你喜欢

转载自blog.csdn.net/weixin_43356538/article/details/114412084