Leetcode 654. 最大二叉树
题目
给定一个不含重复元素的整数数组。一个以此数组构建的最大二叉树定义如下:
- 二叉树的根是数组中的最大元素。
- 左子树是通过数组中最大值左边部分构造出的最大二叉树。
- 右子树是通过数组中最大值右边部分构造出的最大二叉树。
通过给定的数组构建最大二叉树,并且输出这个树的根节点。
示例 :
输入:[3,2,1,6,0,5]
输出:返回下面这棵树的根节点:
6
/ \
3 5
\ /
2 0
\
1
提示:
- 给定的数组的大小在 [1, 1000] 之间。
题解
递归构造
找到当前区间最大元素的下标,取这个数作为根节点,递归构造其左、右子树。详细过程见代码
代码
TreeNode* construct(vector<int>& nums,int l,int r){
if(l > r) return nullptr;
int maxId = l;
for(int i=l+1; i<=r; i++){ //找到区间内最大值的下标
if(nums[i]>nums[maxId]) maxId = i;
}
TreeNode* now = new TreeNode(nums[maxId]);
now->left = construct(nums,l,maxId-1);
now->right = construct(nums,maxId+1,r);
return now;
}
TreeNode* constructMaximumBinaryTree(vector<int>& nums) {
return construct(nums,0,nums.size()-1);
}
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/maximum-binary-tree
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。