#include<iostream>
#include<vector>
using namespace std;
int main() {
ios_base::sync_with_stdio(false);
int n, k;
cin >> n >> k;
int result = 0;
vector<int> road(n);
for(int i = 0; i < n; i++) {
cin >> road[i];
}
for(int i = 1; i < n; i++) {
int dif = road[i-1] - road[i];
if(dif > k) {
road[i] += dif-k;
result += dif-k;
} else if (-dif > k) {
road[i-1] += -dif-k;
result += -dif-k;
}
}
for(int i = n-1; i > 0; i--) {
int dif = road[i-1] - road[i];
if(dif > k) {
road[i] += dif-k;
result += dif-k;
} else if (-dif > k) {
road[i-1] += -dif-k;
result += -dif-k;
}
}
cout << result << endl;
}
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 | #include<iostream> #include<vector> using namespace std; int main() { ios_base::sync_with_stdio(false); int n, k; cin >> n >> k; int result = 0; vector<int> road(n); for(int i = 0; i < n; i++) { cin >> road[i]; } for(int i = 1; i < n; i++) { int dif = road[i-1] - road[i]; if(dif > k) { road[i] += dif-k; result += dif-k; } else if (-dif > k) { road[i-1] += -dif-k; result += -dif-k; } } for(int i = n-1; i > 0; i--) { int dif = road[i-1] - road[i]; if(dif > k) { road[i] += dif-k; result += dif-k; } else if (-dif > k) { road[i-1] += -dif-k; result += -dif-k; } } cout << result << endl; } |
English