给定一个长度为 n
的环形整数数组 nums
,返回 nums
的非空 子数组 的最大可能和 。
环形数组 意味着数组的末端将会与开头相连呈环状。形式上, nums[i]
的下一个元素是 nums[(i + 1) % n]
, nums[i]
的前一个元素是 nums[(i - 1 + n) % n]
。
子数组 最多只能包含固定缓冲区 nums
中的每个元素一次。形式上,对于子数组 nums[i], nums[i + 1], ..., nums[j]
,不存在 i <= k1, k2 <= j
其中 k1 % n == k2 % n
。
输入: nums = [1,-2,3,-2] 输出: 3 解释: 从子数组 [3] 得到最大和 3
输入: nums = [5,-3,5] 输出: 10 解释: 从子数组 [5,5] 得到最大和 5 + 5 = 10
输入: nums = [3,-2,2,-3] 输出: 3 解释: 从子数组 [3] 和 [3,-2,2] 都可以得到最大和 3
n == nums.length
1 <= n <= 3 * 104
-3 * 104 <= nums[i] <= 3 * 104
impl Solution {
pub fn max_subarray_sum_circular(nums: Vec<i32>) -> i32 {
let sum = nums.iter().sum();
let mut max_sum = nums[0];
let mut min_sum = nums[0];
let mut prefix_sum = nums[0];
let mut ret = nums[0].max(sum);
for i in 1..nums.len() {
prefix_sum += nums[i];
ret = ret
.max(prefix_sum - min_sum)
.max(sum - prefix_sum + max_sum);
max_sum = max_sum.max(prefix_sum);
min_sum = min_sum.min(prefix_sum);
}
ret
}
}