#include<iostream>
#include<vector>
using namespace std;
int main() {
int n, k;
cin >> n >> k;
vector<int> arr(n);
for (int i = 0; i < n; i++) {
cin >> arr[i];
}
int gravel = 0;
for (int i = 0; i < n - 1; i++) {
int diff = arr[i + 1] - arr[i];
if (diff > k) {
arr[i] += diff - k;
gravel += diff - k;
} else if (-diff > k) {
arr[i + 1] += - diff - k;
gravel += - diff - k;
}
}
for (int i = n-1; i > 0; i--) {
int diff = arr[i] - arr[i - 1];
if (diff > k) {
arr[i - 1] += diff - k;
gravel += diff - k;
} else if (-diff > k) {
arr[i] += - diff - k;
gravel += - diff - k;
}
}
cout << gravel << 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 | #include<iostream> #include<vector> using namespace std; int main() { int n, k; cin >> n >> k; vector<int> arr(n); for (int i = 0; i < n; i++) { cin >> arr[i]; } int gravel = 0; for (int i = 0; i < n - 1; i++) { int diff = arr[i + 1] - arr[i]; if (diff > k) { arr[i] += diff - k; gravel += diff - k; } else if (-diff > k) { arr[i + 1] += - diff - k; gravel += - diff - k; } } for (int i = n-1; i > 0; i--) { int diff = arr[i] - arr[i - 1]; if (diff > k) { arr[i - 1] += diff - k; gravel += diff - k; } else if (-diff > k) { arr[i] += - diff - k; gravel += - diff - k; } } cout << gravel << endl; return 0; } |
English