题目:
给定一个不含重复元素的整数数组。一个以此数组构建的最大二叉树定义如下:
- 二叉树的根是数组中的最大元素。
- 左子树是通过数组中最大值左边部分构造出的最大二叉树。
- 右子树是通过数组中最大值右边部分构造出的最大二叉树。
通过给定的数组构建最大二叉树,并且输出这个树的根节点。
Example 1:
输入: [3,2,1,6,0,5] 输入: 返回下面这棵树的根节点: 6 / \ 3 5 \ / 2 0 \ 1
注意:
- 给定的数组的大小在 [1, 1000] 之间。
解题思路:
找出最大值,以及最大值的位置,分成左右两个数组,用递归的思路解决。
代码:
class Solution { public: TreeNode* constructMaximumBinaryTree(vector<int>& nums) { if(nums.empty()) return NULL; int mx = INT_MIN, mx_index; for(int i = 0;i < nums.size(); i++){ if(nums[i] > mx){ mx = nums[i]; mx_index = i; } } vector<int> leftArr = vector<int>(nums.begin(), nums.begin()+mx_index); vector<int> rightArr = vector<int>(nums.begin()+mx_index+1, nums.end()); TreeNode* node = new TreeNode(mx); node->left = constructMaximumBinaryTree(leftArr); node->right = constructMaximumBinaryTree(rightArr); return node; } };
相关知识:
c++中的迭代器,非常好用