leetcode 654. 构造最大二叉树 思考分析

题目

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

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

思路和之前的构造树一样：分成左右区间（子树）递归

/**
 * 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* traversal (vector<int>& nums, int numsBegin, int numsEnd) {

        if (numsBegin == numsEnd) return NULL;
        //寻找最大数值
        int maxVal = nums[numsBegin];
        int delimiterIndex=numsBegin;
        for(int i=numsBegin;i<numsEnd;i++)
        {
            if(nums[i]>maxVal)
            {
                delimiterIndex = i;
                maxVal=nums[delimiterIndex];
            }
        }
        TreeNode* root = new TreeNode(maxVal);
        //如果是叶子结点，返回结点
        if (numsEnd - numsBegin == 1) return root;

        // for (delimiterIndex = numsBegin; delimiterIndex < numsEnd; delimiterIndex++) {
        //     if (nums[delimiterIndex] == rootValue) break;
        // }
        // 切割nums数组
        // 左区间，左闭右开[leftInorderBegin, leftInorderEnd)
        int leftorderBegin = numsBegin;
        int leftorderEnd = delimiterIndex;
        // 右区间，左闭右开[rightInorderBegin, rightInorderEnd)
        int rightorderBegin = delimiterIndex + 1;
        int rightorderEnd = numsEnd;

        root->left = traversal(nums, leftorderBegin, leftorderEnd);
        root->right = traversal(nums, rightorderBegin, rightorderEnd);

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