C ++中的K级数最大和
假设我们有一个整数数组arr和一个整数k,我们必须通过重复k次来更改该数组。因此,如果arr=[1,2]且k=3,则修改后的数组将为[1、2、1、2、1、2]。
现在我们必须在修改后的数组中找到最大子数组总和。请注意,子数组的长度可以为0,在这种情况下,其总和为0。由于答案可能非常大,因此求模10^9+7的答案。
因此,如果输入像[1,-2,1]且k=5,则结果将为2。
为了解决这个问题,我们将遵循以下步骤-
定义称为的方法getKadane(),它将采用数组,其工作方式类似于-
ret:=-inf,sum:=0,ret和sum的所有值都是mod10^9+7
对于范围从0到arr的i–1
sum:=arr[i]和arr[i]的最大值+和
ret:=最大ret,总和
如果ret<0,则返回0,否则返回
定义称为的方法getSum(),它将采用数组,其工作方式类似于-
ret:=0,ret的值为mod10^9+7
对于范围从0到arr的i–1
ret:=ret+arr[i]
返回ret
定义称为的方法getPrefix(),它将采用数组,其工作方式类似于-
ret:=-inf,sum:=0,ret和sum的所有值都是mod10^9+7
对于范围从0到arr的i–1
sum:=sum+arr[i]
ret:=ret和sum的最大值
如果ret<0,则返回0,否则返回
定义称为的方法getSuffix(),它将采用数组,其工作方式类似于-
ret:=inf,sum:=0,ret和sum的所有值均为mod10^9+7
对于我在arr范围内的值–1降至0
sum:=sum+arr[i]
ret:=ret和sum的最大值
如果ret<0,则返回0,否则返回
从主要方法中,执行以下操作-
kadane:=getKadane(arr),sum:=getSum(arr),前缀:=getPrefix(arr),后缀:=getSuffix(arr)
如果k为1,则返回kadane
如果sum>1,则返回(sum*(k-2))+前缀+后缀和kadane的最大值
否则返回(prefix+后缀)和kadane的最大值
例子(C++)
让我们看下面的实现以更好地理解-
#include <bits/stdc++.h>
using namespace std;
typedef long long int lli;
const int MOD = 1e9 + 7;
int add(lli a, lli b){
return ((a % MOD) + (b % MOD)) % MOD;
}
int mul(lli a, lli b){
return ((a % MOD) * (b % MOD)) % MOD;
}
class Solution {
public:
int getKadane(vector <int>& arr){
int ret = INT_MIN;
int sum = 0;
for(int i = 0; i < arr.size(); i++){
sum = max(arr[i], arr[i] + sum);
ret = max(ret, sum);
sum %= MOD;
ret %= MOD;
}
return ret < 0? 0 : ret;
}
int getSum(vector <int>& arr){
int ret = 0;
for(int i = 0; i < arr.size(); i++){
ret += arr[i];
ret %= MOD;
}
return ret;
}
int getPrefix(vector <int>& arr){
int ret = INT_MIN;
int sum = 0;
for(int i = 0; i <arr.size(); i++){
sum += arr[i];
sum %= MOD;
ret = max(ret, sum);
ret %= MOD;
}
return ret < 0 ? 0 : ret;
}
int getSuffix(vector <int>& arr){
int sum = 0;
int ret = INT_MIN;
for(int i = arr.size() - 1; i >= 0 ; i--){
sum += arr[i];
ret = max(ret, sum);
sum %= MOD;
ret %= MOD;
}
return ret < 0 ? 0 : ret;
}
int kConcatenationMaxSum(vector<int>& arr, int k) {
int kadane = getKadane(arr);
int sum = getSum(arr);
int prefix = getPrefix(arr);
int suffix = getSuffix(arr);
if(k == 1) return kadane;
if(sum > 0){
return max((int)mul((k-2) , sum) + prefix % MOD + suffix % MOD, kadane);
} else {
return max(add(prefix , suffix), kadane);
}
}
};
main(){
vector<int> v1 = {1,-2,1};
Solution ob;
cout << (ob.kConcatenationMaxSum(v1, 5));
}输入项
[1,-2,1] 5
输出结果
2