给定一个整数数组 nums
,找到一个具有最大和的连续子数组(子数组最少包含一个元素),返回其最大和。
示例:
输入: [-2,1,-3,4,-1,2,1,-5,4],
输出: 6
解释: 连续子数组 [4,-1,2,1] 的和最大,为 6。
进阶:
如果你已经实现复杂度为 O(n) 的解法,尝试使用更为精妙的分治法求解。
思路:
分治法:
将数组一分为2,递归找到左边的最大子序和left_max、右边的最大子序和right_max,再找到跨越中间的最大子序和middle_max,取3者中的最大返回
怎么求跨越中间的最大子序和:
当找到左边的最大子序和时,我们要保留这个最大子序和的长度t1,同理找右边也保留长度t2,那么只需要在中间位置middle往两边扩展min(t1,t2)个空间内找中间最大子序和即可
时间复杂度:O(nlogn)
代码:
class Solution {
int find_cross_max(vector<int>& nums,int left,int middle,int right)
{
int ml=nums[middle],temp=0,mr=nums[middle+1];
for(int i=middle+1;i<=right;i++)
{
temp+=nums[i];
if(mr<temp)
mr=temp;
}
temp=0;
for(int i=middle;i>=left;i--)
{
temp+=nums[i];
if(ml<temp)
ml=temp;
}
return ml+mr;
}
int find_max(vector<int>& nums,int left,int right)
{
if(left==right)
return nums[left];
int a=find_max(nums,left,(left+right)/2);
int b=find_max(nums,(left+right)/2+1,right);
int c=find_cross_max(nums,left,(left+right)/2,right);
int m=a>b?a:b;
m=m>c?m:c;
return m;
}
public:
int maxSubArray(vector<int>& nums) {
return find_max(nums,0,nums.size()-1);
}
};