给定一个不含重复元素的整数数组。一个以此数组构建的最大二叉树定义如下:
二叉树的根是数组中的最大元素。
左子树是通过数组中最大值左边部分构造出的最大二叉树。
右子树是通过数组中最大值右边部分构造出的最大二叉树。
通过给定的数组构建最大二叉树,并且输出这个树的根节点。
示例 :
输入:[3,2,1,6,0,5]
输出:返回下面这棵树的根节点:
6
/ \
3 5
\ /
2 0
\
1
提示:
给定的数组的大小在 [1, 1000] 之间。
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
TreeNode* constructMaximumBinaryTree(vector<int>& nums) {
return constructMaximumBinaryTree2(nums,0,nums.size());
}
TreeNode *constructMaximumBinaryTree2(vector<int> &nums,int start,int end)
{
if(start>=end) return NULL;
int maxIndex=start;
for(int i=start;i<end;i++)
{
if(nums[maxIndex]<nums[i]) maxIndex=i;
//cout<<nums[i]<<" ";
}
TreeNode *p = new TreeNode();
p->val=nums[maxIndex];
p->left=constructMaximumBinaryTree2(nums,start,maxIndex);
p->right=constructMaximumBinaryTree2(nums,maxIndex+1,end);
return p;
}
};
看评论发现和评论里的第一个代码一样。是Java。