1 问题
给定一个不含重复元素的整数数组。一个以此数组构建的最大二叉树定义如下:
二叉树的根是数组中的最大元素。
左子树是通过数组中最大值左边部分构造出的最大二叉树。
右子树是通过数组中最大值右边部分构造出的最大二叉树。
通过给定的数组构建最大二叉树,并且输出这个树的根节点。
示例 :
提示:
给定的数组的大小在 [1, 1000] 之间。
2 解法
最大二叉树的构建过程如下:
/**
* 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:
//left、right分别表示数组的索引
TreeNode* traversal(vector<int>& nums, int left, int right)
{
if(left >= right)
return nullptr;
//寻找最大值在数组中的索引
int maxValIndex = left;
//i从left+1开始,比较nums[i]与nums[maxValIndex]找到最大索引
for(int i = left + 1; i < right; i++)
{
maxValIndex = nums[i] > nums[maxValIndex] ? i : maxValIndex;
}
//根据最大索引切分数组,最大索引处为根节点,左侧为左子树,右侧为右子树
//构造根节点
TreeNode* root = new TreeNode(nums[maxValIndex]);
//构造左子树,左闭右开区间[left, maxValIndex)
root->left = traversal(nums, left, maxValIndex);
//构造右子树,左闭右开区间[maxValIndex + 1, right)
root->right = traversal(nums, maxValIndex + 1, right);
return root;
}
TreeNode* constructMaximumBinaryTree(vector<int>& nums) {
//由于1 <= nums.length <= 1000,无需判断nums = 0的情况
//左闭右开区间[0, nums.size())
return traversal(nums, 0, nums.size());
}
};