题目描述
给定一个不含重复元素的整数数组 nums 。一个以此数组直接递归构建的 最大二叉树 定义如下:
1.二叉树的根是数组 nums 中的最大元素。
2.左子树是通过数组中 最大值左边部分 递归构造出的最大二叉树。
3.右子树是通过数组中 最大值右边部分 递归构造出的最大二叉树。
示例 1:
输入:nums = [3,2,1,6,0,5]
输出:[6,3,5,null,2,0,null,null,1]
示例2:
输入:nums = [3,2,1]
输出:[3,null,2,null,1]
来源:力扣(LeetCode)
一、解题思路
首先理解题目的意思,我们需要用数组中的元素来构建二叉树。既然题目都说到了递归构建二叉树,那么我们可以用递归的方法来构建。
我们可以定义这样一个递归函数来构建二叉树:
首先我们需要遍历当前数组,找出当前数组中最大值的位置和大小,将当前数组最大值赋给二叉树的根节点。之后将最大值左边的数组存入一个新的数组,我们暂将这个新的数组称为left数组。将最大值右边的数组存入一个新的数组,我们暂将这个新的数组称为right数组。在生成left数组和right数组之后,我们还需要定义两个新的二叉树子节点,用来递归生成最大二叉树。有了left和right数组后,递归调用当前这个函数来生成最大二叉树,在当前调用函数开头加上返回条件:若当前数组的长度为0则返回当前树的节点为nullptr。调用函数的参数有两个,一个是指向树节点地址的指针,一个是存放数组元素的容器变量。
二、c++代码实现
/**
* 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:
//定义一个递归调用生成最大二叉树的函数order
void order(vector<int> nums,TreeNode*root){
if(nums.empty()){
root=nullptr;//如果当前数组为空,则当前二叉树子节点返回空节点
return;
}
vector<int> left;
vector <int> right;
int max=nums[0];//当前数组最大值
int index=0;//当前数组最大值索引
for(int i=0;i<nums.size();i++){
if(max<nums[i]){
max=nums[i];
index=i;
}
}
root->val=max;
for(int i=0;i<index;i++){
left.push_back(nums[i]);//构建左边数组
if(i==index-1){
root->left=new TreeNode(0);//构建二叉树左子节点
}
}
for(int i=index+1;i<nums.size();i++){
right.push_back(nums[i]);//构建右边数组
if(i==nums.size()-1){
root->right=new TreeNode(0);//构建二叉树右子节点
}
}
order(left,root->left);
order(right,root->right);
return;
}
TreeNode* constructMaximumBinaryTree(vector<int>& nums) {
if(nums.empty()){
return nullptr;
}
vector <int> left;
vector <int> right;
int max=nums[0];//当前数组最大值
int index=0;//当前数组最大值索引
for(int i=0;i<nums.size();i++){
if(max<nums[i]){
max=nums[i];
index=i;
}
}
TreeNode* root=new TreeNode(max);
for(int i=0;i<index;i++){
left.push_back(nums[i]);//构建左边数组
if(i==index-1){
root->left=new TreeNode(0);//构建二叉树左子节点
}
}
for(int i=index+1;i<nums.size();i++){
right.push_back(nums[i]);//构建右边数组
if(i==nums.size()-1){
root->right=new TreeNode(0);//构建二叉树右子节点
}
}
order(left,root->left);
order(right,root->right);
return root;
}
};
总结
其实本题的解题思路还是比较简单的,关键在于看怎么实现代码能够减少时间复杂度和空间复杂度。对于作者这部分的代码,作者认为在时间复杂度上和空间复杂度上还可以进行提升,由于本人能力和时间有限并没有做优化,只是提供了一种方法。