Leetcode算法训练日记 | day19

一、最大二叉树

1.题目

Leetcode：第 654 题

给定一个不重复的整数数组 nums 。 最大二叉树 可以用下面的算法从 nums 递归地构建:

1. 创建一个根节点，其值为 nums 中的最大值。
2. 递归地在最大值 左边 的 子数组前缀上 构建左子树。
3. 递归地在最大值 右边 的 子数组后缀上 构建右子树。

返回 nums 构建的 最大二叉树 。

示例 1：

输入：nums = [3,2,1,6,0,5]
输出：[6,3,5,null,2,0,null,null,1]
解释：递归调用如下所示：
- [3,2,1,6,0,5] 中的最大值是 6 ，左边部分是 [3,2,1] ，右边部分是 [0,5] 。
    - [3,2,1] 中的最大值是 3 ，左边部分是 [] ，右边部分是 [2,1] 。
        - 空数组，无子节点。
        - [2,1] 中的最大值是 2 ，左边部分是 [] ，右边部分是 [1] 。
            - 空数组，无子节点。
            - 只有一个元素，所以子节点是一个值为 1 的节点。
    - [0,5] 中的最大值是 5 ，左边部分是 [0] ，右边部分是 [] 。
        - 只有一个元素，所以子节点是一个值为 0 的节点。
        - 空数组，无子节点。

示例 2：

输入：nums = [3,2,1]
输出：[3,null,2,null,1]

2.解题思路

首先检查数组的大小，如果只有一个元素，则直接创建一个单节点的二叉树。如果数组有多个元素，函数会找到最大值及其索引，然后创建一个新节点作为根节点，并根据最大值的索引位置，递归地创建左子树和右子树。左子树包含最大值之前的所有元素，右子树包含最大值之后的所有元素。通过这种方式，最终构建出一个最大二叉树。

3.实现代码

#include <iostream>
#include <vector>
using namespace std;

// 定义一个结构体TreeNode，用于表示二叉树的节点。
struct TreeNode {
    int val; // 存储节点的值。
    TreeNode* left; // 指向该节点左子树的指针。
    TreeNode* right; // 指向该节点右子树的指针。
    // TreeNode的构造函数，用于创建一个TreeNode实例。
    // 参数x是节点的值，left和right默认为NULL，表示没有左右子节点。
    TreeNode(int x) : val(x), left(NULL), right(NULL) {}
};

// 根据数组构建最大二叉树。
class Solution {
public:
    // constructMaximumBinaryTree函数用于根据给定的数组nums构建最大二叉树。
    TreeNode* constructMaximumBinaryTree(vector<int>& nums) {
        TreeNode* node = new TreeNode(0);  // 创建一个新的TreeNode节点，初始值为0。
        if (nums.size() == 1) { // 如果数组只有一个元素，直接创建以该元素为根的二叉树并返回。
            node->val = nums[0];
            return node;
        }

        int maxValue = 0; // 初始化最大值和最大值索引。
        int maxValudeIndex = 0;
        for (int i = 0; i < nums.size(); i++) {// 遍历数组，找到最大值及其索引。
            if (nums[i] > maxValue) {
                maxValue = nums[i];
                maxValudeIndex = i;
            }
        }
        node->val = maxValue;// 将找到的最大值赋给当前节点。
        if (maxValudeIndex > 0) {// 如果最大值不在数组的开头，创建左子树。
            vector<int> newVec(nums.begin(), nums.begin() + maxValudeIndex);// 创建一个新数组，包含最大值之前的所有元素。
            node->left = constructMaximumBinaryTree(newVec);// 递归调用构建最大二叉树函数，创建左子树。
        }

        if (maxValudeIndex < nums.size() - 1) {  // 如果最大值之后还有元素，创建右子树。
            vector<int> newVec(nums.begin() + maxValudeIndex + 1, nums.end());// 创建一个新数组，包含最大值之后的所有元素。
            node->right = constructMaximumBinaryTree(newVec); // 递归调用构建最大二叉树函数，创建右子树。
        }
        return node; // 返回构建好的二叉树的根节点。
    }
};

ps：以上皆是本人在探索算法旅途中的浅薄见解，诚挚地希望得到各位的宝贵意见与悉心指导，若有不足或谬误之处，还请多多指教。