【力扣】654. 最大二叉树

题目概述

• 原题链接
  
• 相关节点定义：

/**
 * 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) {}
 * };
 */

思考过程

从以上的示例可以看出，每一次需要找到当前数组（假设为vet）中的最大元素，将其作为tree的根节点，然后将vet划分为三部分

1. 根节点：即最大元素，并记录当前元素下标maxValueIndex，用于划分数组
2. 左半部分：从vet.begin()到最大元素的前一个元素，用程序表述为vector<int> leftpart(vet.begin(),vet.begin() + maxValueIndex)
3. 右半部分：从最大元素下一个至vet.end()，用程序表述为vector<int> rightpart(vet.begin() + maxValueIndex + 1, vet.end())

至此便可用前序遍历构造相应的二叉树，需要注意的是特殊情形：

1. nums数组个数为1的情况，直接构造TreeNode返回即可
2. 根据maxValueIndex进行划分创建新数组时，数组越界问题

代码

class Solution {
public:
    TreeNode* constructMaximumBinaryTree(vector<int>& nums) {
        TreeNode* tree = new TreeNode(0);
        if (nums.size() == 1) {
            tree->val = nums[0];
            return tree;
        }
        int maxValue = -1;
        int maxValueIndex = -1;
        for (int i = 0; i < nums.size(); i++) {
            if (nums[i] > maxValue) {
                maxValue = nums[i];
                maxValueIndex = i;
            }
        }
        tree->val = maxValue;
        if (maxValueIndex > 0) {
            vector<int> leftpart(nums.begin(),nums.begin() + maxValueIndex);
            tree->left = constructMaximumBinaryTree(leftpart);
        }
        if (maxValueIndex < nums.size() - 1) {
            vector<int> rightpart(nums.begin() + maxValueIndex + 1, nums.end());
            tree->right = constructMaximumBinaryTree(rightpart);
        }
        return tree;
    }
};