#ifdef _MSC_VER
#ifndef __GNUC__
#pragma warning(disable: 4996)
#endif
#define main main0
#endif
#include <iostream>
#include <vector>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef unsigned int uint;
int main() {
std::ios_base::sync_with_stdio(false);
std::cin.tie(NULL);
int n, k, max = 0, min = 1000001, imax, imin, wynik = 0;
vector<int> wysokosci;
cin >> n >> k;
wysokosci.resize(n);
for(int i = 0; i < n; ++i) {
int a;
cin >> a;
wysokosci[i] = a;
if(max < a) {
max = a;
imax = i;
}
if(min < a) {
min = a;
imin = i;
}
}
bool zrobione = true;
do {
zrobione = true;
for(int i = 0; i < n - 1; ++i) {
int roznica = wysokosci[i] - wysokosci[i+1] - k;
if(roznica > 0) {
wynik += roznica;
wysokosci[i+1] += roznica;
zrobione = false;
}
}
for(int i = n - 1; i > 0; --i) {
int roznica = wysokosci[i] - wysokosci[i-1] - k;
if(roznica > 0) {
wynik += roznica;
wysokosci[i-1] += roznica;
zrobione = false;
}
}
} while(!zrobione);
cout << wynik << 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 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 | #ifdef _MSC_VER #ifndef __GNUC__ #pragma warning(disable: 4996) #endif #define main main0 #endif #include <iostream> #include <vector> using namespace std; typedef long long ll; typedef unsigned long long ull; typedef unsigned int uint; int main() { std::ios_base::sync_with_stdio(false); std::cin.tie(NULL); int n, k, max = 0, min = 1000001, imax, imin, wynik = 0; vector<int> wysokosci; cin >> n >> k; wysokosci.resize(n); for(int i = 0; i < n; ++i) { int a; cin >> a; wysokosci[i] = a; if(max < a) { max = a; imax = i; } if(min < a) { min = a; imin = i; } } bool zrobione = true; do { zrobione = true; for(int i = 0; i < n - 1; ++i) { int roznica = wysokosci[i] - wysokosci[i+1] - k; if(roznica > 0) { wynik += roznica; wysokosci[i+1] += roznica; zrobione = false; } } for(int i = n - 1; i > 0; --i) { int roznica = wysokosci[i] - wysokosci[i-1] - k; if(roznica > 0) { wynik += roznica; wysokosci[i-1] += roznica; zrobione = false; } } } while(!zrobione); cout << wynik << endl; return 0; } |
English