#ifndef LOCAL
#pragma GCC optimize("O3")
#endif
//#include <bits/stdc++.h>
#include <bit>
#include <print>
#include <iostream>
#include <cassert>
#include <vector>
#include <map>
#include <unordered_map>
#include <set>
#include <algorithm>
#include <functional>
#include <ranges>
#include <numeric>
#define FOR(i,p,k) for(int i=(p); i<=(k); ++i)
#define REP(i,k) FOR(i,0,(k)-1)
#define RFOR(i,p,n) for(int i=(p); i>=(n); --i)
#define all(x) (x).begin(), (x).end()
#define rall(x) (x).rbegin(), (x).rend()
#define ssize(x) int((x).size())
#define fi first
#define se second
#define V vector
#define pb push_back
#define eb emplace_back
#define C const
#define pn printf("\n")
using namespace std;
typedef long long ll;
typedef V <int> vi;
typedef V <ll> vll;
typedef C int ci;
typedef C ll cll;
typedef pair <int, int> pii;
typedef pair <ll, ll> pll;
void chmin(auto &a, auto b){a=min(a,b);}
void chmax(auto &a, auto b){a=max(a,b);}
ci inf=2.1e9;
cll infll=4.5e18;
int I(){
int z;
while ((z=getchar_unlocked())<'-'||z>'9');
C bool czymin=z=='-';
int r=czymin ? 0 : z-'0';
while ((z=getchar_unlocked())>='0'&&z<='9')
r=r*10+z-'0';
return czymin ? -r : r;
}
void ans(){
ci n=I(),k=I();
vi wys(n); generate(all(wys), I);
ll wyn=0;
auto czy=[&](ci i) {
return abs(wys[i]-wys[i-1])>k;
};
auto fix=[&](ci i) {
if (wys[i]<wys[i-1]) {
wyn+=wys[i-1]-k-wys[i];
wys[i]=wys[i-1]-k;
}
else {
wyn+=wys[i]-k-wys[i-1];
wys[i-1]=wys[i]-k;
}
};
FOR(i, 1, n-1) {
RFOR(j, i, 1) {
if (!czy(j))
break;
fix(j);
}
}
printf("%lld\n", wyn);
}
int main(){
ios_base::sync_with_stdio(0),cin.tie(0);
int tt=1;
//tt=I();
while (tt--)ans();
}
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 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 | #ifndef LOCAL #pragma GCC optimize("O3") #endif //#include <bits/stdc++.h> #include <bit> #include <print> #include <iostream> #include <cassert> #include <vector> #include <map> #include <unordered_map> #include <set> #include <algorithm> #include <functional> #include <ranges> #include <numeric> #define FOR(i,p,k) for(int i=(p); i<=(k); ++i) #define REP(i,k) FOR(i,0,(k)-1) #define RFOR(i,p,n) for(int i=(p); i>=(n); --i) #define all(x) (x).begin(), (x).end() #define rall(x) (x).rbegin(), (x).rend() #define ssize(x) int((x).size()) #define fi first #define se second #define V vector #define pb push_back #define eb emplace_back #define C const #define pn printf("\n") using namespace std; typedef long long ll; typedef V <int> vi; typedef V <ll> vll; typedef C int ci; typedef C ll cll; typedef pair <int, int> pii; typedef pair <ll, ll> pll; void chmin(auto &a, auto b){a=min(a,b);} void chmax(auto &a, auto b){a=max(a,b);} ci inf=2.1e9; cll infll=4.5e18; int I(){ int z; while ((z=getchar_unlocked())<'-'||z>'9'); C bool czymin=z=='-'; int r=czymin ? 0 : z-'0'; while ((z=getchar_unlocked())>='0'&&z<='9') r=r*10+z-'0'; return czymin ? -r : r; } void ans(){ ci n=I(),k=I(); vi wys(n); generate(all(wys), I); ll wyn=0; auto czy=[&](ci i) { return abs(wys[i]-wys[i-1])>k; }; auto fix=[&](ci i) { if (wys[i]<wys[i-1]) { wyn+=wys[i-1]-k-wys[i]; wys[i]=wys[i-1]-k; } else { wyn+=wys[i]-k-wys[i-1]; wys[i-1]=wys[i]-k; } }; FOR(i, 1, n-1) { RFOR(j, i, 1) { if (!czy(j)) break; fix(j); } } printf("%lld\n", wyn); } int main(){ ios_base::sync_with_stdio(0),cin.tie(0); int tt=1; //tt=I(); while (tt--)ans(); } |
English