力扣654 最大二叉树

题目:

给定一个不重复的整数数组 nums 。 最大二叉树 可以用下面的算法从 nums 递归地构建:
    创建一个根节点,其值为 nums 中的最大值。
    递归地在最大值 左边 的 子数组前缀上 构建左子树。
    递归地在最大值 右边 的 子数组后缀上 构建右子树。
返回 nums 构建的 最大二叉树 。

示例:

输入: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 的节点。
        - 空数组,无子节点。

思路:

其实和《从中序与后序遍历序列构造二叉树》思路相似。

class Solution {
    public TreeNode constructMaximumBinaryTree(int[] nums) {
        //1.空节点
        if(nums.length==0){
            return null;
        }
        //2.只有根节点
        if (nums.length == 1) {
            TreeNode root=new TreeNode(nums[0]);
            return root;
        }
        //3.找到最大值作为切割点
        int maxIndex=0;
        int max=nums[0];//注意这里max要放外面
        for(int i=0;i<nums.length;i++){
            if(nums[i]>max){
                max=nums[i];
                maxIndex=i;//最大值
            }
        }
        TreeNode root=new TreeNode(nums[maxIndex]);//构建节点

        //4.切割出左区间
        int[] left=new int[maxIndex];
        for(int i=0;i<maxIndex;i++){
            left[i]=nums[i];
        }
        //5.切割出右区间
        int[] right=new int[nums.length-maxIndex-1];
        for(int i=maxIndex+1;i<nums.length;i++){
            right[i-maxIndex-1]=nums[i];
        }
        //6.递归处理左右区间
        root.left=constructMaximumBinaryTree(left);
        root.right=constructMaximumBinaryTree(right);

        return root;
    }
}

 

 

posted @ 2023-02-02 23:46  壹索007  阅读(7)  评论(0编辑  收藏  举报