-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathmaximum_sum_circular_subarray.py
51 lines (35 loc) · 1.48 KB
/
maximum_sum_circular_subarray.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
"""
Given a circular integer array nums of length n, return the maximum possible sum of a non-empty subarray of nums.
A circular array means the end of the array connects to the beginning of the array. Formally, the next element of nums[i] is nums[(i + 1) % n] and the previous element of nums[i] is nums[(i - 1 + n) % n].
A subarray may only include each element of the fixed buffer nums at most once. Formally, for a subarray nums[i], nums[i + 1], ..., nums[j], there does not exist i <= k1, k2 <= j with k1 % n == k2 % n.
Example 1:
Input: nums = [1,-2,3,-2]
Output: 3
Explanation: Subarray [3] has maximum sum 3.
Example 2:
Input: nums = [5,-3,5]
Output: 10
Explanation: Subarray [5,5] has maximum sum 5 + 5 = 10.
Example 3:
Input: nums = [-3,-2,-3]
Output: -2
Explanation: Subarray [-2] has maximum sum -2.
Constraints:
n == nums.length
1 <= n <= 3 * 104
-3 * 104 <= nums[i] <= 3 * 104
"""
class Solution:
def maxSubarraySumCircular(self, nums: List[int]) -> int:
total_sum = 0
current_max_sum = 0
current_min_sum = 0
max_sum = -math.inf
min_sum = math.inf
for i in nums:
total_sum = total_sum + i
current_max_sum = max(current_max_sum + i, i)
current_min_sum = min(current_min_sum + i, i)
max_sum = max(max_sum, current_max_sum)
min_sum = min(min_sum, current_min_sum)
return max_sum if max_sum < 0 else max(max_sum, total_sum - min_sum)