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 <bits/stdc++.h>
#include "homecoming.h"
using namespace std;
const long long INF = 1e16;
const int MAX_N = 2e6;
vector<long long> dp[MAX_N + 1];
long long solve( int n, int k, int a[], int b[] ) {
long long ans = 0;
for ( int i = 0; i <= n + 1; i++ )
dp[i].resize( k + 1 );
for ( int take = 0; take <= 1; take++ ) {
if ( !take ) {
dp[n][0] = 0;
for ( int j = 1; j <= k; j++ )
dp[n][j] = dp[n][j - 1] - b[j - 1];
} else {
for ( int j = 0; j <= k; j++ )
dp[n][j] = 0;
}
for ( int i = n - 1; i >= 0; i-- ) {
dp[i][0] = -INF;
if ( i >= k * take ) {
for ( int j = 0; j <= k; j++ )
dp[i][0] = max( dp[i][0], dp[i + 1][j] );
}
for ( int j = 1; j < k; j++ )
dp[i][j] = dp[i + 1][j - 1] - b[i];
dp[i][k] = max( dp[i + 1][k - 1], dp[i + 1][k] ) - b[i] + a[i];
}
for ( int j = 0; j <= k; j++ )
ans = max( ans, dp[0][j] );
}
return ans;
}
# | 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... |