题解:https://leetcode-cn.com/problems/maximum-binary-tree/solution/zui-da-er-cha-shu-by-leetcode/
复杂度分析
时间复杂度:O(n^2) 。方法 dfs 一共被调用 n 次。每次递归寻找根节点时,需要遍历当前索引范围内所有元素找出最大值。一般情况下,每次遍历的复杂度为 O(logn),总复杂度为 O(nlogn)。最坏的情况下,数组 numsnums 有序,总的复杂度为 O(n^2) 。
空间复杂度:O(n)。递归调用深度为 n。平均情况下,长度为 n 的数组递归调用深度为O(logn)。
/**
* * 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) {
if(nums.size() == 0) {
return nullptr;
}
return dfs(nums, 0, nums.size()-1);
}
TreeNode* dfs(const vector<int>& nums, int begin, int end) {
if(begin < 0 || begin >= nums.size() || end < 0 || end >= nums.size() || begin > end) {
return nullptr;
}
int max_value = nums[begin];
int max_index = begin;
for(int i = begin; i <= end; ++i) {
if(max_value < nums[i]) {
max_value = nums[i];
max_index = i;
}
}
TreeNode* new_node = new TreeNode(max_value);
new_node->left = dfs(nums, begin, max_index-1);
new_node->right = dfs(nums, max_index+1, end);
return new_node;
}
};