天天看点

leetcode 918. 环形子数组的最大和

给定一个由整数数组 A 表示的环形数组 C,求 C 的非空子数组的最大可能和。

在此处,环形数组意味着数组的末端将会与开头相连呈环状。(形式上,当0 <= i < A.length 时 C[i] = A[i],而当 i >= 0 时 C[i+A.length] = C[i])

此外,子数组最多只能包含固定缓冲区 A 中的每个元素一次。(形式上,对于子数组 C[i], C[i+1], …, C[j],不存在 i <= k1, k2 <= j 其中 k1 % A.length = k2 % A.length)

示例 1:

输入:[1,-2,3,-2]

输出:3

解释:从子数组 [3] 得到最大和 3

示例 2:

输入:[5,-3,5]

输出:10

解释:从子数组 [5,5] 得到最大和 5 + 5 = 10

示例 3:

输入:[3,-1,2,-1]

输出:4

解释:从子数组 [2,-1,3] 得到最大和 2 + (-1) + 3 = 4

示例 4:

输入:[3,-2,2,-3]

输出:3

解释:从子数组 [3] 和 [3,-2,2] 都可以得到最大和 3

示例 5:

输入:[-2,-3,-1]

输出:-1

解释:从子数组 [-1] 得到最大和 -1

提示:

-30000 <= A[i] <= 30000

1 <= A.length <= 30000

来源:力扣(LeetCode)

链接:https://leetcode-cn.com/problems/maximum-sum-circular-subarray

著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

分析:设整个数组的和为sum, 连续子数组的最大和为maxSum, 最小和为minSum, 那最终的结果res=max(maxSum, sum-minSum), sum-minSum表示那种中间最小,首尾各一段加起来和最大的情况。

class Solution(object):
    def maxSubarraySumCircular(self, A):
        """
        :type A: List[int]
        :rtype: int
        """
        res=A[0]
        minRes=A[0]
        curMax=A[0]        #表示到当前位置为止的连续最大和
        curMin=A[0]        #表示到当前位置为止的连续最小和
        sum=A[0]
        for i in range(1,len(A)):
            if curMax>=0:
                curMax+=A[i]
            else:
                curMax=A[i]
            if curMin<0:
                curMin+=A[i]
            else:
                curMin=A[i]
            sum+=A[i]
            if curMax>res:
                res=curMax
            if curMin<minRes:
                minRes=curMin
        if res<sum-minRes and sum-minRes>0:
            res=sum-minRes
        return res