#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef long double ld;
#define st first
#define nd second
void solve() {
int n, k; cin >> n >> k;
vector<int> a(n);
for (int i = 0; i < n; i++) cin >> a[i];
ll ans = 0;
for (int i = 1; i < n; i++) {
int x = max(a[i], a[i-1]-k);
ans += x - a[i];
a[i] = x;
for (int j = i-1; j >= 0; j--) {
x = max(a[j], a[j+1]-k);
ans += x-a[j];
a[j] = x;
}
}
cout << ans << '\n';
}
int main()
{
ios_base::sync_with_stdio(false);
cin.tie(0);
int t = 1;
// cin >> t;
while (t--) {
solve();
}
}
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 | #include <bits/stdc++.h> using namespace std; typedef long long ll; typedef long double ld; #define st first #define nd second void solve() { int n, k; cin >> n >> k; vector<int> a(n); for (int i = 0; i < n; i++) cin >> a[i]; ll ans = 0; for (int i = 1; i < n; i++) { int x = max(a[i], a[i-1]-k); ans += x - a[i]; a[i] = x; for (int j = i-1; j >= 0; j--) { x = max(a[j], a[j+1]-k); ans += x-a[j]; a[j] = x; } } cout << ans << '\n'; } int main() { ios_base::sync_with_stdio(false); cin.tie(0); int t = 1; // cin >> t; while (t--) { solve(); } } |
English