题目描述:
给定一个不含重复元素的整数数组。一个以此数组构建的最大二叉树定义如下:
- 二叉树的根是数组中的最大元素。
- 左子树是通过数组中最大值左边部分构造出的最大二叉树。
- 右子树是通过数组中最大值右边部分构造出的最大二叉树。
通过给定的数组构建最大二叉树,并且输出这个树的根节点。
链接:https://leetcode-cn.com/problems/maximum-binary-tree
思路:每次先找到数组中最大的值,左子树是当前最大值的左边,右子树是当前最大值的右边。
- 找到数组最大值,构造节点
- 左子树在当前最大值左边
- 右子树在当前最大值右边
- 上述方法进行递归
- 递归出口:两端值相等的时候
代码:
class Solution {
public:
int SearchMaxValPos(vector<int>& nums,int start,int end)//找寻最大值的位置
{
int maxval = nums[start];
int pos = start;
for(int i = start + 1;i < end;++i)
{
if(nums[i] > maxval)
{
pos = i;
maxval = nums[i];
}
}
return pos;
}
TreeNode* constructMaxi(vector<int>& nums,int start,int end)
{
if(start == end)//递归出口
return nullptr;
TreeNode* root = new TreeNode;
int max = SearchMaxValPos(nums,start,end);//最大值所在位置
root->val = nums[max];
root->left = constructMaxi(nums,start,max);//左子树在左边
root->right = constructMaxi(nums,max + 1, end);//右子树在右边
return root;//构造完毕 返回
}
TreeNode* constructMaximumBinaryTree(vector<int>& nums) {
if(nums.empty())//为空,直接返回
return nullptr;
return constructMaxi(nums,0,nums.size());//调用该函数,进行返回
}
};