#include <bits/stdc++.h>
using namespace std;
long long solve () {
long long n, k;
cin >> n >> k;
vector<long long> a(n);
for (long long i = 0; i < n; i++) {
cin >> a[i];
}
vector<long long> pokrycie = a;
for (long long i = 0; i < n; i++) {
// Left:
long long cur = a[i] - k;
for (long long j = i + 1; j < n; j++) {
pokrycie[j] = max(pokrycie[j], cur);
cur -= k;
}
// Right:
cur = a[i] - k;
for (long long j = i - 1; j >= 0; j--) {
pokrycie[j] = max(pokrycie[j], cur);
cur -= k;
}
}
long long ans = 0;
for (long long i = 0; i < n; i++) {
ans += pokrycie[i] - a[i];
}
return ans;
}
int main() {
ios_base::sync_with_stdio(false);
cin.tie(NULL);
cout << solve() << endl;
return 0;
}
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 | #include <bits/stdc++.h> using namespace std; long long solve () { long long n, k; cin >> n >> k; vector<long long> a(n); for (long long i = 0; i < n; i++) { cin >> a[i]; } vector<long long> pokrycie = a; for (long long i = 0; i < n; i++) { // Left: long long cur = a[i] - k; for (long long j = i + 1; j < n; j++) { pokrycie[j] = max(pokrycie[j], cur); cur -= k; } // Right: cur = a[i] - k; for (long long j = i - 1; j >= 0; j--) { pokrycie[j] = max(pokrycie[j], cur); cur -= k; } } long long ans = 0; for (long long i = 0; i < n; i++) { ans += pokrycie[i] - a[i]; } return ans; } int main() { ios_base::sync_with_stdio(false); cin.tie(NULL); cout << solve() << endl; return 0; } |
English