#include <iostream>
#include <vector>
using namespace std;
int main()
{
int n;
int k;
cin >> n >> k;
vector<int> sciezka(n);
for (int i = 0; i < n; i++) {
cin >> sciezka[i];
}
//for (int i = 0; i < n; i++) {
// cout << sciezka[i] << " ";
//}
//cout << endl;
long long zwir = 0;
for (int i = 0; i < n - 1; i++) {
if (abs(sciezka[i] - sciezka[i + 1]) <= k) continue;
if (sciezka[i] > sciezka[i + 1]) {
int taczka = sciezka[i] - sciezka[i + 1] - k;
zwir += taczka;
sciezka[i + 1] += taczka;
}
else {
int taczka = sciezka[i + 1] - sciezka[i] - k;
zwir += taczka;
sciezka[i] += taczka;
int j = i - 1;
while (j >= 0 && sciezka[j+1] - sciezka[j] > k) {
taczka = sciezka[j+1] - sciezka[j] - k;
zwir += taczka;
sciezka[j] += taczka;
j--;
}
}
}
//for (int i = 0; i < n; i++) {
// cout << sciezka[i] << " ";
//}
//cout << endl;
cout << zwir << 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 42 43 44 45 46 47 48 49 50 51 52 | #include <iostream> #include <vector> using namespace std; int main() { int n; int k; cin >> n >> k; vector<int> sciezka(n); for (int i = 0; i < n; i++) { cin >> sciezka[i]; } //for (int i = 0; i < n; i++) { // cout << sciezka[i] << " "; //} //cout << endl; long long zwir = 0; for (int i = 0; i < n - 1; i++) { if (abs(sciezka[i] - sciezka[i + 1]) <= k) continue; if (sciezka[i] > sciezka[i + 1]) { int taczka = sciezka[i] - sciezka[i + 1] - k; zwir += taczka; sciezka[i + 1] += taczka; } else { int taczka = sciezka[i + 1] - sciezka[i] - k; zwir += taczka; sciezka[i] += taczka; int j = i - 1; while (j >= 0 && sciezka[j+1] - sciezka[j] > k) { taczka = sciezka[j+1] - sciezka[j] - k; zwir += taczka; sciezka[j] += taczka; j--; } } } //for (int i = 0; i < n; i++) { // cout << sciezka[i] << " "; //} //cout << endl; cout << zwir << endl; } |
English