二叉树算法题（13）最大二叉树

目录

最大二叉树

描述

示例 1

示例 2

提示

方法：递归

---

最大二叉树

描述

给定一个不含重复元素的整数数组 nums 。一个以此数组直接递归构建的 最大二叉树 定义如下：

• 二叉树的根是数组 nums 中的最大元素
• 左子树是通过数组中 最大值左边部分 递归构造出的最大二叉树
• 右子树是通过数组中 最大值右边部分 递归构造出的最大二叉树

返回有给定数组 nums 构建的 最大二叉树 。

示例 1

输入：nums = [3,2,1,6,0,5]
输出：[6,3,5,null,2,0,null,null,1]
解释：递归调用如下所示：
- [3,2,1,6,0,5] 中的最大值是 6 ，左边部分是 [3,2,1] ，右边部分是 [0,5] 。
    - [3,2,1] 中的最大值是 3 ，左边部分是 [] ，右边部分是 [2,1] 。
        - 空数组，无子节点。
        - [2,1] 中的最大值是 2 ，左边部分是 [] ，右边部分是 [1] 。
            - 空数组，无子节点。
            - 只有一个元素，所以子节点是一个值为 1 的节点。
    - [0,5] 中的最大值是 5 ，左边部分是 [0] ，右边部分是 [] 。
        - 只有一个元素，所以子节点是一个值为 0 的节点。
        - 空数组，无子节点。

示例 2

输入：nums = [3,2,1]
输出：[3,null,2,null,1]

提示

• 1 <= nums.length <= 1000
• 0 <= nums[i] <= 1000
• nums 中的所有整数 互不相同

方法：递归

这个题其实比前面的利用中序后序数组构造二叉树要简单，本题思路是先找到最大值的下标，然后分为两个数组递归构造。

class Solution {
    public TreeNode constructMaximumBinaryTree(int[] nums) {
        return reCur(nums,0,nums.length);
    }

    public int maxIndex(int[] nums,int low,int high){
        int max=nums[low];
        int index=low;
        for (int i = low; i < high; i++) {
            if (nums[i]>max){
                max=nums[i];
                index=i;
            }
        }
        return index;
    }

    public TreeNode reCur(int nums[],int low,int high){
        if (low==high) return null;//如果越界，直接返回null
        int index=maxIndex(nums,low,high);//找到当前要构造的根节点下标（即最大值）
        TreeNode root=new TreeNode(nums[index]);//创建树节点
        root.left=reCur(nums,low,index);//递归创建左子树
        root.right=reCur(nums,index+1,high);//递归创建右子树
        return root;
    }
}