给定一个不含重复元素的整数数组。一个以此数组构建的最大二叉树定义如下:
- 二叉树的根是数组中的最大元素。
- 左子树是通过数组中最大值左边部分构造出的最大二叉树。
- 右子树是通过数组中最大值右边部分构造出的最大二叉树。
-
通过给定的数组构建最大二叉树,并且输出这个树的根节点。
-
Example 1:
输入: [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) {
int nodeNum = nums.size();
TreeNode* root = NULL;
findMax(nums, 0, nodeNum, root);
return root;
}
void findMax(vector<int>& nums, int start, int end, TreeNode* &root){
if(end <= start) return;
int max = -999999;
int max_index = -1;
vector<int>::iterator iter_start = nums.begin();
vector<int>::iterator iter;
for(int i = start; i < end; i++){
iter = iter_start + i;
if(*iter > max){
max = *iter;
max_index = i;
}
}
root = new TreeNode(max);//新建对象
findMax(nums, start, max_index, root->left);
findMax(nums, max_index+1, end, root->right);
}
};
欢迎各位评论!!!