#include <bits/stdc++.h>
#include <utility>
using namespace std;
#define endl "\n"
#define ll long long
void solve() {
int n, k;
cin >> n >> k;
vector<int> a(n);
for(int i = 0; i < n; i++) {
cin >> a[i];
}
vector<int> upleft(n, 0);
vector<int> upright(n, 0);
int top = a[0];
//first left-to-right
for(int i = 1; i < n; i++) {
top -= k;
upleft[i] = max(top - a[i], 0);
top = max(top, a[i]);
}
// right-to-left
top = a[n-1];
for(int i = n-2; i >= 0; i--) {
top -= k;
upright[i] = max(top - a[i], 0);
top = max(top, a[i]);
}
int ans = 0;
for(int i = 0; i < n; i++) {
ans += max(upleft[i], upright[i]);
}
cout << ans << endl;
}
int main() {
ios::sync_with_stdio(false);
cin.tie(nullptr);
cout.tie(nullptr);
int t = 1;
//cin >> t;
while (t--) {
solve();
}
}
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 45 46 47 48 49 50 51 52 53 54 55 | #include <bits/stdc++.h> #include <utility> using namespace std; #define endl "\n" #define ll long long void solve() { int n, k; cin >> n >> k; vector<int> a(n); for(int i = 0; i < n; i++) { cin >> a[i]; } vector<int> upleft(n, 0); vector<int> upright(n, 0); int top = a[0]; //first left-to-right for(int i = 1; i < n; i++) { top -= k; upleft[i] = max(top - a[i], 0); top = max(top, a[i]); } // right-to-left top = a[n-1]; for(int i = n-2; i >= 0; i--) { top -= k; upright[i] = max(top - a[i], 0); top = max(top, a[i]); } int ans = 0; for(int i = 0; i < n; i++) { ans += max(upleft[i], upright[i]); } cout << ans << endl; } int main() { ios::sync_with_stdio(false); cin.tie(nullptr); cout.tie(nullptr); int t = 1; //cin >> t; while (t--) { solve(); } } |
English