LeetCode-654-构建最大二叉树

Posted 2022-12-08 herongweiV

LeetCode 654 最大二叉树

链接：https://leetcode-cn.com/problems/maximum-binary-tree/

【题目】 给定一个不含重复元素的整数数组 nums 。一个以此数组直接递归构建的 最大二叉树 定义如下：

• 二叉树的根是数组 nums 中的最大元素。
• 左子树是通过数组中 最大值左边部分 递归构造出的最大二叉树。
• 右子树是通过数组中 最大值右边部分 递归构造出的最大二叉树。
• 返回有给定数组 nums 构建的 最大二叉树 。

【思路】

• 构造树一般采用的是前序遍历，因为先构造中间节点，然后递归构造左子树和右子树。
• 确定递归函数的参数和返回值：参数就是传入的是存放元素的数组，返回该数组构造的二叉树的头结点，返回类型是指向节点的指针。
• 确定终止条件：题目中说了输入的数组大小一定是大于等于1的，所以我们不用考虑小于1的情况，那么当递归遍历的时候，如果传入的数组大小为1，说明遍历到了叶子节点了。
• 确定单层递归的逻辑：优化思路，就是每次分隔不用定义新的数组，而是通过下标索引直接在原数组上操作。

/**
 * 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) 
        return traversal(nums, 0, nums.size());
    
    TreeNode* traversal(std::vector<int>& nums, int left, int right) 
        if (left >= right) return nullptr;

        int max_val_idx = left;
        for (int i = left + 1; i < right; ++i) 
            if (nums[i] > nums[max_val_idx]) 
                max_val_idx = i;
            
        
        TreeNode* root = new TreeNode(nums[max_val_idx]);
        root->left  = traversal(nums, left, max_val_idx);
        root->right = traversal(nums, max_val_idx + 1, right);
        return root;
    
;

本文由博客一文多发平台 OpenWrite 发布！