利用数组构建最大二叉树(根节点大于左右子树)

leetcode 654. 最大二叉树

题目链接

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

二叉树的根是数组 nums 中的最大元素。
左子树是通过数组中 最大值左边部分 递归构造出的最大二叉树。
右子树是通过数组中 最大值右边部分 递归构造出的最大二叉树。
返回有给定数组 nums 构建的 最大二叉树 。

递归三部曲

1.参数和返回值：数组 vector< int >以及他的起始点 int;输出树的根节点
2.单层递归逻辑：找数组最大的值为当前根节点，数组左边的为根节点的左子树，右边的为右子树
3.终止条件：当前数组只有一个元素返回当前节点，没有元素返回空

class Solution {
    TreeNode* bulidTree(vector<int>& nums, int begin, int end)
    {
        if(begin == end)
            return nullptr;
        int maxNum = nums[begin];
        int index = begin;
        for(int i = begin + 1; i < end; ++i)
        {
            if(nums[i] > maxNum)
            {
                index = i;
                maxNum = nums[i];
            }
        }
        TreeNode* root = new TreeNode(maxNum);
        if(end - begin == 1)
            return root;
        root->left = bulidTree(nums, begin, index);
        root->right = bulidTree(nums,index + 1, end);
        return root;
    }
public:
    TreeNode* constructMaximumBinaryTree(vector<int>& nums) {
       return bulidTree(nums, 0, nums.size());
    }
};