#include <bits/stdc++.h>
using namespace std;
int main() {
int n;
long long k;
cin >> n >> k;
vector<long long> a(n);
vector<long long> h(n);
for (int i=0; i<n; i++)
{
cin>>a[i];
h[i]=a[i];
}
bool z=true;
while (z)
{
z = false;
for (int i=1; i<n; i++)
{
if (h[i]<h[i-1]-k)
{
h[i]=h[i-1]-k;
z=true;
}
}
for (int i=n-2; i>=0; i--)
{
if (h[i]<h[i+1]-k)
{
h[i]=h[i+1]-k;
z=true;
}
}
}
long long res=0;
for (int i=0; i<n; i++)
{
res+=(h[i]-a[i]);
}
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 | #include <bits/stdc++.h> using namespace std; int main() { int n; long long k; cin >> n >> k; vector<long long> a(n); vector<long long> h(n); for (int i=0; i<n; i++) { cin>>a[i]; h[i]=a[i]; } bool z=true; while (z) { z = false; for (int i=1; i<n; i++) { if (h[i]<h[i-1]-k) { h[i]=h[i-1]-k; z=true; } } for (int i=n-2; i>=0; i--) { if (h[i]<h[i+1]-k) { h[i]=h[i+1]-k; z=true; } } } long long res=0; for (int i=0; i<n; i++) { res+=(h[i]-a[i]); } cout<<res; return 0; } |
English