leetcode 654最大二叉树

力扣链接：654. 最大二叉树

题目：

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

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

通过给定的数组构建最大二叉树，并且输出这个树的根节点。

Example 1:

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

      6
    /   \
   3     5
    \    / 
     2  0   
       \
        1

注意:

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

答案：

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    TreeNode* constructMaximumBinaryTree(vector<int> &nums, int start, int stop)
    {
        
        if(stop == start)
            return nullptr;
            
        auto max_idx = max_element(nums.begin()+start, nums.begin()+stop);
        size_t idx = distance(nums.begin(), max_idx);
        TreeNode *root = new TreeNode(*max_idx);
        root->left = constructMaximumBinaryTree(nums, start, idx);
        root->right = constructMaximumBinaryTree(nums, idx + 1, stop);
        return root;
    }

    TreeNode* constructMaximumBinaryTree(vector<int>& nums) {
        int start = 0, stop = nums.size();
        TreeNode *root = constructMaximumBinaryTree(nums, start, stop);
        return root;
    }
};