#include <bits/stdc++.h>
#include <unordered_map>
using namespace std;
typedef long long ll;
typedef vector<int> vi;
typedef vector<ll> vll;
typedef pair<int, int> pii;
typedef vector<vi> vvi;
typedef vector<bool> vb;
#define eb(x) emplace_back(x)
#define loop(i,a,b) for (int (i) = (a); (i) < b; (i)++)
#define rloop(i,a,b) for (int (i) = (a); (i) >= b; (i)--)
int main()
{
cin.tie(0)->sync_with_stdio(false);
ll n, k, wyn = 0;
cin >> n >> k;
vi ciag(n);
loop(i, 0, n) cin >> ciag[i];
loop(i, 1, n)
{
if (ciag[i] < ciag[i - 1])
{
int zmiana = max(ciag[i - 1] - ciag[i] - k, 0ll);
ciag[i] += zmiana;
wyn += zmiana;
}
else
{
int zmiana = max(ciag[i] - ciag[i - 1] - k, 0ll);
if (zmiana)
{
ciag[i - 1] += zmiana;
wyn += zmiana;
rloop(j, i - 1, 1)
{
int zmiana2 = max(ciag[j] - ciag[j - 1] - k, 0ll);
if (!zmiana2) break;
ciag[j - 1] += zmiana2;
wyn += zmiana2;
}
}
}
}
cout << wyn;
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 | #include <bits/stdc++.h> #include <unordered_map> using namespace std; typedef long long ll; typedef vector<int> vi; typedef vector<ll> vll; typedef pair<int, int> pii; typedef vector<vi> vvi; typedef vector<bool> vb; #define eb(x) emplace_back(x) #define loop(i,a,b) for (int (i) = (a); (i) < b; (i)++) #define rloop(i,a,b) for (int (i) = (a); (i) >= b; (i)--) int main() { cin.tie(0)->sync_with_stdio(false); ll n, k, wyn = 0; cin >> n >> k; vi ciag(n); loop(i, 0, n) cin >> ciag[i]; loop(i, 1, n) { if (ciag[i] < ciag[i - 1]) { int zmiana = max(ciag[i - 1] - ciag[i] - k, 0ll); ciag[i] += zmiana; wyn += zmiana; } else { int zmiana = max(ciag[i] - ciag[i - 1] - k, 0ll); if (zmiana) { ciag[i - 1] += zmiana; wyn += zmiana; rloop(j, i - 1, 1) { int zmiana2 = max(ciag[j] - ciag[j - 1] - k, 0ll); if (!zmiana2) break; ciag[j - 1] += zmiana2; wyn += zmiana2; } } } } cout << wyn; return 0; } |
English