python--lintcode42.最大子数组II

描述

给定一个整数数组,找出两个 不重叠 子数组使得它们的和最大。
每个子数组的数字在数组中的位置应该是连续的。
返回最大的和。

子数组最少包含一个数

您在真实的面试中是否遇到过这个题?  是

样例

给出数组 [1, 3, -1, 2, -1, 2]
这两个子数组分别为 [1, 3] 和 [2, -1, 2] 或者 [1, 3, -1, 2] 和 [2],它们的最大和都是 7

挑战

要求时间复杂度为 O(n)

这一题的解决要依赖于上一题,上一题见:https://blog.csdn.net/wenqiwenqi123/article/details/81586084

那么这一题怎么来做呢?当然我们仍然用动态规划,时间复杂度O(n),分三步走:

先从左遍历数组,保存遍历到每个元素的当前的最大连续子数组的值;从右遍历数组,记录同样的最大连续子数组的值;遍历数组比较每种组合的左右最大连续子数组的和值,返回最大的值。

代码如下:

class Solution:
    """
    @param: nums: A list of integers
    @return: An integer denotes the sum of max two non-overlapping subarrays
    """
    def maxTwoSubArrays(self, nums):
        # write your code here
        if(len(nums)==2):return nums[0]+nums[1]
        # dp数组记录每个位置结尾的最大和
        # dpleft从左到右遍历
        dpleft=[0  for i in range(len(nums))]
        maxp = -99999
        sum = 0
        for i in range(0, len(nums)):
            if (sum < 0):
                sum = nums[i]
            else:
                sum += nums[i]
            maxp = max(maxp, sum)
            dpleft[i] = maxp

        # dpright从右到左遍历
        dpright=[0  for i in range(len(nums))]
        maxp = -99999
        sum = 0
        for i in range(len(nums)-1,-1,-1):
            if (sum < 0):
                sum = nums[i]
            else:
                sum += nums[i]
            maxp = max(maxp, sum)
            dpright[i] = maxp

        # dpleft里存的是每个位置左边子数组最大和,dpright里存的是每个位置右边子数组最大和
        # 所以最后统计一遍两者之和最大值
        maxp=-99999999
        for i in range(1,len(dpleft)):
            sum=dpleft[i-1]+dpright[i]
            if(sum>maxp):
                maxp=sum
        return maxp



s=Solution()
print(s.maxTwoSubArrays([1, 3, -1, 2, -1, 2]))

猜你喜欢

转载自blog.csdn.net/wenqiwenqi123/article/details/81631894