题目链接:https://leetcode-cn.com/problems/maximum-binary-tree/
题目描述
给定一个不含重复元素的整数数组。一个以此数组构建的最大二叉树定义如下:
二叉树的根是数组中的最大元素。
左子树是通过数组中最大值左边部分构造出的最大二叉树。
右子树是通过数组中最大值右边部分构造出的最大二叉树。
通过给定的数组构建最大二叉树,并且输出这个树的根节点。
示例 :
输入:[3,2,1,6,0,5]
输出:返回下面这棵树的根节点:
6
/ \
3 5
\ /
2 0
\
1
提示:
给定的数组的大小在 [1, 1000] 之间。
思路
递归
代码
// 递归
class Solution {
public:
TreeNode* constructMaximumBinaryTree(vector<int>& nums) {
if(nums.empty()) return nullptr;
return helper(nums,0,nums.size()-1);
}
TreeNode* helper(vector<int>& nums, int begin, int end){
if(begin > end)
return nullptr;
int maxNum = INT_MIN;
int index = begin;
// 找最大值
for(int i = begin; i<= end; ++i){
if(maxNum < nums[i]){
maxNum = nums[i];
index = i;
}
}
TreeNode* root = new TreeNode(maxNum);
root->left = helper(nums,begin,index-1); // 递归构建左子树
root->right = helper(nums,index+1,end); // 递归构建右子树
return root;
}
};