Leetcode 654. 最大二叉树 C++

Leetcode 654. 最大二叉树

题目

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

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

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

示例 ：

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

      6
    /   \
   3     5
    \    / 
     2  0   
       \
        1

提示：

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

题解

递归构造
找到当前区间最大元素的下标，取这个数作为根节点，递归构造其左、右子树。详细过程见代码

代码

	TreeNode* construct(vector<int>& nums,int l,int r){
        if(l > r)   return nullptr;
        int maxId = l;
        for(int i=l+1; i<=r; i++){		//找到区间内最大值的下标
            if(nums[i]>nums[maxId]) maxId = i;
        }
        TreeNode* now = new TreeNode(nums[maxId]);
        now->left = construct(nums,l,maxId-1);
        now->right = construct(nums,maxId+1,r);
        return now;
    }
    TreeNode* constructMaximumBinaryTree(vector<int>& nums) {
        return construct(nums,0,nums.size()-1);
    }

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/maximum-binary-tree
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。