题目描述:
给定一个不含重复元素的整数数组。一个以此数组构建的最大二叉树定义如下:
二叉树的根是数组中的最大元素。
左子树是通过数组中最大值左边部分构造出的最大二叉树。
右子树是通过数组中最大值右边部分构造出的最大二叉树。
通过给定的数组构建最大二叉树,并且输出这个树的根节点。
示例 :
输入:[3,2,1,6,0,5]
输出:返回下面这棵树的根节点:
提示:
给定的数组的大小在 [1, 1000] 之间。
方法1:
主要思路:
(1)使用递归建立树结构,每次在数组中找出给定范围内的最大值,使用该最大值建立当前根节点,再以该节点左边的结点建立其左子树,该结点右边的结点建立其右子树,最后返回根节点即可;
/**
* 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* dfs(vector<int>& nums,int left,int right){
//终止条件
if(left>right){
return NULL;
}
//找出当前范围内的最大值的位置
int max_index=left;
for(int i=left+1;i<=right;++i){
if(nums[max_index]<nums[i]){
max_index=i;
}
}
//以最大值建立当前根节点
TreeNode* root=new TreeNode(nums[max_index]);
//以最大值左右两边的各自范围内的值,建立左右子树
root->left=dfs(nums,left,max_index-1);
root->right=dfs(nums,max_index+1,right);
//返回根节点
return root;
}
TreeNode* constructMaximumBinaryTree(vector<int>& nums) {
return dfs(nums,0,nums.size()-1);
}
};