Lc 1191
Contents
Leetcode 1191, couldn’t get the solution
Leetcode 1191
Given an integer array arr
and an integer k
, modify the array by repeating it k
times.
For example, if arr = [1, 2]
and k = 3
then the modified array will be [1, 2, 1, 2, 1, 2]
.
Return the maximum sub-array sum in the modified array. Note that the length of the sub-array can be 0
and its sum in that case is 0
.
As the answer can be very large, return the answer modulo 10^9 + 7
.
Example 1:
Input: arr = [1,2], k = 3
Output: 9
Example 2:
Input: arr = [1,-2,1], k = 5
Output: 2
Example 3:
Input: arr = [-1,-2], k = 7
Output: 0
Constraints:
1 <= arr.length <= 10^5
1 <= k <= 10^5
-10^4 <= arr[i] <= 10^4
Solution
dp
We can try if k = 1, k = 2, k > 2; and see whether the sum of array is >= 0 or < 0.
typedef long long LL;
const int MOD = 1e9 + 7;
class Solution {
public:
int kConcatenationMaxSum(vector<int>& arr, int k) {
// l max prefix, r max suffix, s means across the array max
LL mx = 0, l = 0, r = 0, sum = 0, s = 0;
for (int i = 0; i < arr.size(); i ++ ) {
sum += arr[i];
l = max(l, sum);
s = max(s, 0ll) + arr[i];
mx = max(mx, s);
if (i + 1 == arr.size()) r = s;
}
if (k == 1) return mx % MOD;
if (sum < 0) return max(mx, l + r) % MOD;
return max(sum * (LL)(k - 2) + l + r, mx) % MOD;
}
};