LeetCode 654 最大二叉树
链接:leetcode-cn.com/problems/ma…
【题目】 给定一个不含重复元素的整数数组 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 发布!