leetcode 53. 最大子序和 Python

题目

给定一个整数数组 nums ，找到一个具有最大和的连续子数组（子数组最少包含一个元素），返回其最大和，地址。
示例：

输入: [-2,1,-3,4,-1,2,1,-5,4],
输出: 6
解释: 连续子数组 [4,-1,2,1] 的和最大，为 6。

解题思路

动态规划，分成2种情况考虑：

1. 加上当前的数结果小于0，则可以都舍弃并把当前的值与记录的最大值比较取最大。
2. 如果加上当前的数结果大于等于0，则可以保留当前的值继续往后加，并且要把加完以后的值与记录的最大值作比较取最大。
3. 特别注意：把结果最开始赋值无穷小，因为有类似的[-2]测试用例，可以保证在取max的时候能取到-2而不是0。

Python实现

class Solution:
    def maxSubArray(self, nums: List[int]) -> int:
        
        res, sum_nums = float("-inf"), 0
        for i in range(len(nums)):
            sum_nums += nums[i]
            if sum_nums >= 0:
                sum_nums = sum_nums
                res = max(res, sum_nums)
            else:
                res = max(res, sum_nums)
                sum_nums = 0
            
            
        return res