This submission is migrated from previous version of oj.uz, which used different machine for grading. This submission may have different result if resubmitted.
#include<homecoming.h>
#include<bits/stdc++.h>
using namespace std;
const long long inf=1e18;
long long solve(int n, int k, int* a, int*b1){
vector<int> b(n*2);
for(int i=0; i<n; i++){
b[i]=b[i+n]=b1[i];
}
vector<int> prefixsum(n*2+1, 0);
for(int i=1; i<=n*2; i++){
prefixsum[i]=prefixsum[i-1]+b[i-1];
}
vector<long long> dp(n+1, -inf);
long long ans=0;
dp[0]=0;
for(int i=1; i<=n; i++){
for(int j=1; j<=k; j++){
if(i-j<0) continue;
int cost=prefixsum[i+k-2]-prefixsum[i-j+k-2];
dp[i]=max(dp[i], dp[i-j]+a[i-1]-cost);
ans=max(ans, dp[i]);
}
}
return ans;
}
/*signed main(){
ios_base::sync_with_stdio(false);
cin.tie(0);
int n, k;
cin >> n >> k;
vector<int> a(n);
vector<int> b(n);
for(int i=0; i<n; i++){
cin >> a[i];
}
for(int i=0; i<n; i++){
cin >> b[i];
}
cout << solve(n, k, a, b) << "\n";
}*/
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |