给定一个整数数组 nums ,找到一个具有最大和的连续子数组(子数组最少包含一个元素),返回其最大和。
示例:
输入: [-2,1,-3,4,-1,2,1,-5,4],
输出: 6
解释: 连续子数组 [4,-1,2,1] 的和最大,为 6。
思路:
首先当给定数组中全部元素都为负数的时候,最大子序和就是数组中最大的那个元素;当数组中元素有正有负的时候,最大子序和一定是大于等于零的。因为找的是连续子数组的最大和,所以在遍历过程中需要有两个变量来记录和,其中一个是local_max,他在遍历过程中会不断更新,如果得到的子序和小于零,就更新为零,然后重新开始求和,如果还是大于零的,就继续遍历,继续加和;global_max记录的是遍历过程中的最大子序和,最后返回global_max
class Solution(object):
def maxSubArray(self, nums):
"""
:type nums: List[int]
:rtype: int
"""
# 动态规划
if max(nums)<0:
return max(nums)
local_max = 0
global_max = 0
for num in nums:
local_max = max(0,local_max+num)
global_max = max(local_max,global_max)
return global_max
分治法
class Solution:
def maxSubArray(self, nums: List[int]) -> int:
#分治法
return self.maxSubPart(nums, 0, len(nums)-1)
def maxSubPart(self, nums, left, right):
if left == right:
return nums[left]
mid = (left + right) //2
return max(self.maxSubPart(nums, left, mid),
self.maxSubPart(nums, mid+1, right),
self.maxSubAll(nums, left, mid, right))
def maxSubAll(self, nums, left, mid, right):
tmp, left_sum = 0, -float('inf')
for i in range(mid, left-1, -1):
tmp += nums[i]
if tmp > left_sum:
left_sum = tmp
tmp, right_sum = 0, -float('inf')
for i in range(mid+1, right+1):
tmp += nums[i]
if tmp>right_sum:
right_sum = tmp
return left_sum + right_sum