【leetcode】654 最大二叉树（二叉树）

题目链接：https://leetcode-cn.com/problems/maximum-binary-tree/

题目描述

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

二叉树的根是数组中的最大元素。
左子树是通过数组中最大值左边部分构造出的最大二叉树。
右子树是通过数组中最大值右边部分构造出的最大二叉树。
通过给定的数组构建最大二叉树，并且输出这个树的根节点。

示例 ：

输入：[3,2,1,6,0,5]
输出：返回下面这棵树的根节点：

      6
    /   \
   3     5
    \    / 
     2  0   
       \
        1

提示：

给定的数组的大小在 [1, 1000] 之间。

思路

递归

代码

// 递归
class Solution {
public:
    TreeNode* constructMaximumBinaryTree(vector<int>& nums) {
        if(nums.empty()) return nullptr;
        return helper(nums,0,nums.size()-1);
    }
    
    TreeNode* helper(vector<int>& nums, int begin, int end){
        if(begin > end)
            return nullptr;
        int maxNum = INT_MIN;
        int index = begin;
        
        // 找最大值
        for(int i = begin; i<= end; ++i){
            if(maxNum < nums[i]){
                maxNum = nums[i];
                index = i;
            }
        }
        
        TreeNode* root = new TreeNode(maxNum);
        root->left = helper(nums,begin,index-1);    // 递归构建左子树
        root->right = helper(nums,index+1,end);     // 递归构建右子树
        return root;
    }
};