leetcode最大子序和(python)
题目
给定一个整数数组 nums ,找到一个具有最大和的连续子数组(子数组最少包含一个元素),返回其最大和。
示例:
输入: [-2,1,-3,4,-1,2,1,-5,4]
输出: 6
解释: 连续子数组 [4,-1,2,1] 的和最大,为 6。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/maximum-subarray
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解题思路
由题可知,需要找到一个最大和的连续子数组,并返回其最大和。既然是一个和,那么我们就可以用两个变量来表示:max_sum
和current_sum
,其中max_sum
用来保存最终的值,而current_sum
用来作比较的,就相当于一个。最开始,两个变量都保存列表的第一个元素的值,然后通过遍历元素,当我现在保存的和的值比现在元素的值小时,那么就舍弃原来的保存的值,换成那个较大的元素的值,然后就比较我们保存的最终的值与现在的那个值,若我们保存最终的和小于现在的那个数的和,那么就更新它的值。其中代码current_sum = max(current_time + num, num)
的作用非常重要,首先current_sum + num
表示求的是一个子列表的和,而num
表示当前的元素,当前一个子列表的和小于改元素,我们就更新current_sum
为num
的值,然后开始下一个子列表。整个函数的最关键代码就是这一句。
代码
class Solution(object):
def maxSubArray(self, nums):
"""
:type nums: List[int]
:rtype: int
"""
if len(nums) == 0:
print('invalid array list')
return
max_sum = current_sum = nums[0]
for num in nums[1:]:
current_sum = max(current_sum + num, num)
max_sum = max(current_sum, max_sum)
return max_sum
Solution().maxSubArray([-2,1,-3,4,-1,2,1,-5,4])