leetcode刷题：最大二叉树

---

题目描述

    给定一个不含重复元素的整数数组 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;
    }
};

 

总结

   其实本题的解题思路还是比较简单的，关键在于看怎么实现代码能够减少时间复杂度和空间复杂度。对于作者这部分的代码，作者认为在时间复杂度上和空间复杂度上还可以进行提升，由于本人能力和时间有限并没有做优化，只是提供了一种方法。