title
给定一个由整数数组 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] <= 300001 <= A.length <= 30000
solution
分成两端考虑,对每个i,找到i+1及之后的rightmost
class Solution:
def maxSubarraySumCircular(self, nums: List[int]) -> int:
minv = float('-inf')
ans = minv
cur = 0
for num in nums:
cur = num + max(cur, 0)
ans = max(ans, cur)
right = [0 for _ in range(len(nums) + 1)]
for i in range(len(nums) - 1, -1, -1):
right[i] = right[i + 1] + nums[i]
print(right)
maxright = [0 for _ in range(len(nums) + 1)]
maxright[len(nums)] = minv
for i in range(len(nums) - 1, -1, -1):
maxright[i] = max(maxright[i + 1], right[i])
cur = 0
for i in range(len(nums)):
cur += nums[i]
ans = max(ans, cur + max(0, maxright[i + 1]))
return ans