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;
#define intl long long
intl solve(int n, int k, int *A, int *B) {
vector <intl> a, b, pref(n, 0);
function <intl(intl)> get=[&](intl idx) {
if(idx + k - 1 < n) return (pref[idx+k-1]-(idx?pref[idx-1]:0LL))*1LL;
else {
intl ans = (pref[n-1]-(idx?pref[idx-1]:0LL)) * 1LL;
ans += (pref[(idx+k-1)%n])*1LL;
return ans;
}
};
for(intl i = 0; i < n; i ++) a.push_back(A[i]);
for(intl i = 0; i < n; i ++) b.push_back(B[i]);
intl ans = 0;
/*for(int i = 0; i < n; i ++) {
if(i) pref[i] = pref[i-1];
pref[i] += (b[i] * 1LL);
}*/
// for(auto i : pref) cout << i << " ";
// cout << "\n";
for(intl i = 0; i < n; i ++) {
intl bank = a[i], kk = k;
for(intl j = i; kk >= 1; j ++, kk --) {
j %= n;
bank -= b[j];
}
if(bank < 0) continue;
ans += bank;
}
assert(ans >= 0);
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... |