题目:
给定一个不含重复元素的整数数组 nums 。一个以此数组直接递归构建的 最大二叉树 定义如下:
二叉树的根是数组 nums 中的最大元素。
左子树是通过数组中 最大值左边部分 递归构造出的最大二叉树。
右子树是通过数组中 最大值右边部分 递归构造出的最大二叉树。
返回有给定数组 nums 构建的 最大二叉树
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode() : val(0), left(nullptr), right(nullptr) {}
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
* };
*/
class Solution {
public:
TreeNode* constructMaximumBinaryTree(vector<int>& nums) {
// 本题在最后反了一个很蠢的错误,在递归的地方,把 right 写成了 left
// 真的就是好蠢好蠢哦!!!!!!!!!!!!!!
if(nums.empty())
{
return nullptr;
}
int delimIndex=0,max=nums[0];
for(int i =1; i<nums.size(); i++)
{
if(nums[i]>max)
{
// 这里可不是交换哦~~
// swap(nums[delimIndex],max);
max = nums[i];
delimIndex = i;
}
}
TreeNode* root = new TreeNode(max);
if(nums.size() == 1)
{
return root;
}
// 这里 学到一个点 如何取出 vector中的部分元素,用 {} 进行列表初始化
// 这里构造左右子树时,没有判断,因为至少要保证 区间中有一个数值
if(delimIndex > 0)
{
vector<int> left{nums.begin(),nums.begin()+delimIndex};
root->left = constructMaximumBinaryTree(left);
}
if(delimIndex <(nums.size()-1))
{
// 用 {} 进行列表初始化
vector<int> right{nums.begin()+delimIndex+1,nums.end()};
root->right = constructMaximumBinaryTree(right);
}
return root;
}
};