题目:
给定一个不含重复元素的整数数组 nums 。一个以此数组直接递归构建的 最大二叉树 定义如下:
二叉树的根是数组 nums 中的最大元素。
左子树是通过数组中 最大值左边部分 递归构造出的最大二叉树。
右子树是通过数组中 最大值右边部分 递归构造出的最大二叉树。
示例:
解法
构建二叉树一般用的前序,因为先构建根节点
一般情况来说:如果让空节点(空指针)进入递归,就不加if,如果不让空节点进入递归,就加if限制一下, 终止条件也会相应的调整。
代码
class Solution {
public:
TreeNode* traversal(vector<int>& nums, int left, int right) {
if (left >= right) return nullptr;
//分割,找到局部最大值
int delimitIndex = left;
for (int i = left + 1; i < right; i++) {
if (nums[i] > nums[delimitIndex]) {
delimitIndex = i;
}
}
//创建节点,
TreeNode* node = new TreeNode(nums[delimitIndex]);
//左右节点返回值递归
node->left = traversal(nums, left, delimitIndex);
node->right = traversal(nums, delimitIndex + 1, right);
return node;
}
TreeNode* constructMaximumBinaryTree(vector<int>& nums) {
return traversal(nums, 0, nums.size());
}
};