#include <bits/stdc++.h>
using namespace std;
int main()
{
ios_base::sync_with_stdio(0);
cin.tie(0);
int n;
long long k;
cin>>n>>k;
vector<long long> a(n);
for(int i=0; i<n; i++)
{
cin>>a[i];
}
if(n==0)
{
cout<<0<<endl;
return 0;
}
vector<long long> h=a;
for(int i=1; i<n; i++)
{
if(h[i]<h[i-1]-k)
{
h[i]=h[i-1]-k;
}
}
for(int i=n-2; i>=0; i--)
{
if(h[i]<h[i+1]-k)
{
h[i]=h[i+1]-k;
}
}
long long total=0;
for(int i=0; i<n; i++)
{
total+=(h[i]-a[i]);
}
cout<<total<<endl;
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 | #include <bits/stdc++.h> using namespace std; int main() { ios_base::sync_with_stdio(0); cin.tie(0); int n; long long k; cin>>n>>k; vector<long long> a(n); for(int i=0; i<n; i++) { cin>>a[i]; } if(n==0) { cout<<0<<endl; return 0; } vector<long long> h=a; for(int i=1; i<n; i++) { if(h[i]<h[i-1]-k) { h[i]=h[i-1]-k; } } for(int i=n-2; i>=0; i--) { if(h[i]<h[i+1]-k) { h[i]=h[i+1]-k; } } long long total=0; for(int i=0; i<n; i++) { total+=(h[i]-a[i]); } cout<<total<<endl; return 0; } |
English