#include <bits/stdc++.h>
#include <bits/extc++.h>
using namespace std;
using namespace __gnu_pbds;
#ifdef DEBUG
template<typename T1,typename T2>auto& operator<<(ostream&o,pair<T1,T2>a){return o<<"("<<a.first<<", "<<a.second<<")";}
template<typename T,size_t...I>void pt(ostream&o,T t,index_sequence<I...>){o<<"(";(...,(o<<(I?", ":"")<< get<I>(t)));o<<")";}
template<typename...A>auto& operator<<(ostream&o,tuple<A...>t){pt(o,t,index_sequence_for<A...>{});return o;}
template<typename T,typename O>auto& operator<<(O&o,T a){o<<"{";for(auto b:a)o<<b<<", ";return o<<"}";}
#define db(x...) cerr << "\033[92m" << "[" #x "]: ", [](auto... args) { ((cerr << args << ", "),...) << "\033[0m" << "\n"; }(x)
#else
#define db(...)
#endif
#define sz(x) ((int)(x).size())
#define all(x) (x).begin(), (x).end()
#define F first
#define S second
template<class T>
using iset = tree<T, null_type, less<T>, rb_tree_tag, tree_order_statistics_node_update>;
using ll = long long;
using ld = long double;
using pll = pair<ll,ll>;
using vi = vector<int>;
mt19937 mrand(random_device{}());
ll rnd(ll l, ll r) { return l + mrand() % (r - l + 1);}
int main() {
cin.tie(0)->sync_with_stdio(0); cin.exceptions(cin.failbit);
ll n, k;
cin >> n >> k;
vector<ll> v(n);
for(ll i = 0 ; i < n ; ++i) cin >> v[i];
std::set<pll> pq;
for(ll i = 0 ; i < n ; ++i) pq.emplace(v[i], i);
ll res = 0;
while(!pq.empty()) {
auto [val, u] = *pq.rbegin();
pq.erase({val, u});
if(u > 0 && v[u-1] < val) {
res += max(0ll, val-v[u-1]-k);
pq.erase({v[u-1], u-1});
v[u-1] = max(v[u-1], val-k);
pq.emplace(v[u-1], u-1);
}
if(u < n-1 && v[u+1] < val) {
res += max(0ll, val-v[u+1]-k);
pq.erase({v[u+1], u+1});
v[u+1] = max(v[u+1], val-k);
pq.emplace(v[u+1], u+1);
}
}
cout << res;
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 61 62 63 64 | #include <bits/stdc++.h> #include <bits/extc++.h> using namespace std; using namespace __gnu_pbds; #ifdef DEBUG template<typename T1,typename T2>auto& operator<<(ostream&o,pair<T1,T2>a){return o<<"("<<a.first<<", "<<a.second<<")";} template<typename T,size_t...I>void pt(ostream&o,T t,index_sequence<I...>){o<<"(";(...,(o<<(I?", ":"")<< get<I>(t)));o<<")";} template<typename...A>auto& operator<<(ostream&o,tuple<A...>t){pt(o,t,index_sequence_for<A...>{});return o;} template<typename T,typename O>auto& operator<<(O&o,T a){o<<"{";for(auto b:a)o<<b<<", ";return o<<"}";} #define db(x...) cerr << "\033[92m" << "[" #x "]: ", [](auto... args) { ((cerr << args << ", "),...) << "\033[0m" << "\n"; }(x) #else #define db(...) #endif #define sz(x) ((int)(x).size()) #define all(x) (x).begin(), (x).end() #define F first #define S second template<class T> using iset = tree<T, null_type, less<T>, rb_tree_tag, tree_order_statistics_node_update>; using ll = long long; using ld = long double; using pll = pair<ll,ll>; using vi = vector<int>; mt19937 mrand(random_device{}()); ll rnd(ll l, ll r) { return l + mrand() % (r - l + 1);} int main() { cin.tie(0)->sync_with_stdio(0); cin.exceptions(cin.failbit); ll n, k; cin >> n >> k; vector<ll> v(n); for(ll i = 0 ; i < n ; ++i) cin >> v[i]; std::set<pll> pq; for(ll i = 0 ; i < n ; ++i) pq.emplace(v[i], i); ll res = 0; while(!pq.empty()) { auto [val, u] = *pq.rbegin(); pq.erase({val, u}); if(u > 0 && v[u-1] < val) { res += max(0ll, val-v[u-1]-k); pq.erase({v[u-1], u-1}); v[u-1] = max(v[u-1], val-k); pq.emplace(v[u-1], u-1); } if(u < n-1 && v[u+1] < val) { res += max(0ll, val-v[u+1]-k); pq.erase({v[u+1], u+1}); v[u+1] = max(v[u+1], val-k); pq.emplace(v[u+1], u+1); } } cout << res; return 0; } |
English