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 = 1e17;
long long S[2000010];
long long dp[2][2000010];
long long solve(int N, int K, int *A, int *B) {
long long all = 0;
for(int i = 0; i < N; i++) {
all += A[i] - B[i];
}
long long ans = all;
long long sum = 0;
for(int i = 0; i < K; i++) {
sum += B[i];
}
for(int i = 0; i < N; i++) {
S[i] = sum;
sum += B[(i + K) % N] - B[i];
}
dp[0][0] = 0;
dp[1][0] = -inf;
for(int j = 1; j < N; j++) {
int id = j;
int prev = (id + N - 1) % N;
dp[0][id] = max(dp[0][prev], dp[1][prev]);
dp[1][id] = max(dp[0][prev] + A[id] - S[id], dp[1][prev] + A[id] - B[(id + K - 1) % N]);
}
ans = max(ans, dp[0][N - 1]);
ans = max(ans, dp[1][N - 1]);
dp[0][0] = -inf;
dp[1][0] = A[0] - S[0];
for(int j = 1; j < N; j++) {
int id = j;
int prev = (id + N - 1) % N;
dp[0][id] = max(dp[0][prev], dp[1][prev]);
dp[1][id] = max(dp[0][prev] + A[id] - S[id], dp[1][prev] + A[id] - B[(id + K - 1) % N]);
}
ans = max(ans, dp[0][N - 1]);
dp[0][0] = -inf;
dp[1][0] = A[0] - B[K - 1];
for(int j = 1; j < N; j++) {
int id = j;
int prev = (id + N - 1) % N;
dp[0][id] = max(dp[0][prev], dp[1][prev]);
dp[1][id] = max(dp[0][prev] + A[id] - S[id], dp[1][prev] + A[id] - B[(id + K - 1) % N]);
}
ans = max(ans, dp[1][N - 1]);
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... |