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)
思路: 同最大连续和,只是变成环型,分两种,第一种同非环,第二种考虑头尾情况,此时反过来求最小连续和,然后sum-min得到最大值,不过要考虑全为负数情况,此时返回最小负数
class Solution:
def maxSubarraySumCircular(self, nums: List[int]) -> int:
max_sum = cur_max = nums[0]
min_sum = cur_min = nums[0]
for n in nums[1:]:
cur_max = max(cur_max + n, n)
cur_min = min(cur_min + n, n)
max_sum = max(cur_max, max_sum)
min_sum = min(cur_min, min_sum)
if max_sum < 0:
return max_sum
return max(max_sum, sum(nums) - min_sum)