面试题42. 连续子数组的最大和
输入一个整型数组,数组里有正数也有负数。数组中的一个或连续多个整数组成一个子数组。求所有子数组的和的最大值。
要求时间复杂度为O(n)。
示例1:
输入: nums = [-2,1,-3,4,-1,2,1,-5,4]
输出: 6
解释: 连续子数组 [4,-1,2,1] 的和最大,为 6。
思路:动态规划
时间复杂度 O(N) : 线性遍历数组 nums 即可获得结果,使用 O(N) 时间。
class Solution:
def maxSubArray(self, nums: List[int]) -> int:
length = len(nums)
dp=[0]*len(nums)
dp[0]=nums[0]
for i in range(1,length):
if dp[i-1] < 0:
dp[i] = nums[i]
else:
dp[i] = dp[i-1]+nums[i]
return max(dp)
提升
由于 dp[i]只与 dp[i-1]和 nums[i] 有关系,因此可以将原数组 nums 用作 dp 列表,即直接在 nums 上修改即可。
由于省去 dp 列表使用的额外空间,因此空间复杂度从 O(N) 降至 O(1) 。
class Solution:
def maxSubArray(self, nums: List[int]) -> int:
for i in range(1, len(nums)):
nums[i] += max(nums[i - 1], 0)
return max(nums)
来源:力扣leetcode