题目(leecode T654):
给定一个不重复的整数数组 nums 。 最大二叉树 可以用下面的算法从 nums 递归地构建:
nums 中的最大值。返回 nums 构建的 最大二叉树 。

方法:本题可以看出是按照同一个处理原则的重复处理,因此还是用递归法比较方便。确定三要素
1:传入参数和返回值,因为给定的是一个数组。所以我们传入的参数就是数组,而我们要返回的是一个构造好的树,因此返回值是存储树节点的指针
2:确定终止条件,因为题目中已经表示数组不会为空,因此当数组元素只剩一个时,代表该元素是树的叶子节点,到这里就可以结束了。
3:确定单层的处理逻辑,每一层中我们要处理的是,我们要找到当数组中最大的元素,用该元素来构造根节点,然后该元素左半部分继续递归生成左子树,右半部分继续递归生成右子树。
题解:
- class Solution {
- public:
- TreeNode* constructMaximumBinaryTree(vector<int>& nums) {
- TreeNode* node = new TreeNode(0);
- if(nums.size() == 1){
- node -> val = nums[0];
- return node;
- }
- int maxValue = 0; //找数组最大的元素当根节点
- int maxValueIndex = 0;
- for(int i = 0; i < nums.size(); i++){
- if(nums[i] > maxValue){
- maxValue = nums[i];
- maxValueIndex = i;
- }
- }
- node->val = maxValue;
- if(maxValueIndex > 0){ //切分左子树
- vector<int> newVec(nums.begin(), nums.begin() + maxValueIndex);
- node -> left = constructMaximumBinaryTree(newVec); //递归生成左子树
- }
- if(maxValueIndex < (nums.size() - 1)){ //切分右子树
- vector<int> newVec(nums.begin() + maxValueIndex + 1, nums.end());
- node ->right = constructMaximumBinaryTree(newVec); //递归生成右子树
- }
- return node;
- }
- };