一、题目:
https://leetcode.cn/problems/maximum-subarray/description/?envType=study-plan-v2&envId=top-100-liked
给你一个整数数组 nums
,请你找出一个具有最大和的连续子数组(子数组最少包含一个元素),返回其最大和。
子数组 是数组中的一个连续部分。
示例 1:
输入:nums = [-2,1,-3,4,-1,2,1,-5,4] 输出:6 解释:连续子数组 [4,-1,2,1] 的和最大,为 6 。
示例 2:
输入:nums = [1] 输出:1
二、思路
前缀和 与 最大子串:方法本来适用于股票买入卖出日子的选择。
https://leetcode.cn/problems/best-time-to-buy-and-sell-stock/submissions/484917997/
本题要找最大值,可以化为最大的累加和减去最小的累加和,反正前缀一样。
list(accumulate(nums,initial=0))为从0位置开始的各个累加和。
三、代码
class Solution:
def maxSubArray(self, nums: List[int]) -> int:
mi, ma, ans = 0, 0, max(nums)
lst=list(accumulate(nums, initial = 0))
for x in lst[1:]:
if x < mi:
mi, ma = x, x + ans
elif x > ma:
ma, ans = x, x - mi
return ans