连续子数组的最大和
题目
输入一个整型数组,数组里有正数也有负数。数组中的一个或连续多个整数组成一个子数组。求所有子数组的和的最大值。
要求时间复杂度为 O ( n ) O(n) O(n)。
思路 动态规划:
设f[i]表示以第i个数字结尾时的最大子数组和,因为题目要求连续子数组,那么:
C++
class Solution {
public:
int maxSubArray(vector<int>& nums) {
int length = nums.size();
int f[length];
f[0] = nums[0];
int res = nums[0];
for(int i=1; i<length; ++i)
{
if(f[i-1]<0)
f[i]=nums[i];
else
f[i] = nums[i]+f[i-1];
if(res<f[i])
res = f[i];
}
return res;
}
};
python
class Solution:
def maxSubArray(self, nums: List[int]) -> int:
res = nums[0]
f = [-100] * len(nums)
f[0] = nums[0]
for i in range(1, len(nums)):
if f[i-1] <0:
f[i] = nums[i]
else:
f[i] = f[i-1]+nums[i]
if res < f[i]:
res = f[i]
return res