leetcode 654. 最大二叉树

目录

• 题目描述：
• 示例:
• 解法：

题目描述：

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

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

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

示例:

输入: [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* createTree(vector<int>& nums, int l, int r){
        if(l == r){
            return new TreeNode(nums[l]);
        }else if(l > r){
            return NULL;
        }else{
            int val = nums[l];
            int mid = l;
            for(int i = l+1; i <= r; i++){
                if(val < nums[i]){
                    val = nums[i];
                    mid = i;
                }
            }
            TreeNode* root = new TreeNode(val);
            root->left = createTree(nums, l, mid-1);
            root->right = createTree(nums, mid+1, r);
            return root;
        }
    }
    
    TreeNode* constructMaximumBinaryTree(vector<int>& nums) {
        int sz = nums.size();
        int l = 0, r = sz-1;
        return createTree(nums, l, r);
    }
};