#include<bits/stdc++.h>
using namespace std;
int main(){
ios::sync_with_stdio(false);
cin.tie(0); cout.tie(0);
int n,k; cin>>n>>k;
vector<int> a(n);
for(auto &i:a)cin>>i;
long long c=0;
while(1){
long long d=0;
for(int i=1;i<n;i++){
if(a[i]<a[i-1]-k)d+=a[i-1]-k-a[i],a[i]=a[i-1]-k;
else if(a[i]>a[i-1]+k)d+=a[i]-k-a[i-1],a[i-1]=a[i]-k;
}
if(d)c+=d;
else break;
}
cout<<c<<endl;
return 0;
}
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 | #include<bits/stdc++.h> using namespace std; int main(){ ios::sync_with_stdio(false); cin.tie(0); cout.tie(0); int n,k; cin>>n>>k; vector<int> a(n); for(auto &i:a)cin>>i; long long c=0; while(1){ long long d=0; for(int i=1;i<n;i++){ if(a[i]<a[i-1]-k)d+=a[i-1]-k-a[i],a[i]=a[i-1]-k; else if(a[i]>a[i-1]+k)d+=a[i]-k-a[i-1],a[i-1]=a[i]-k; } if(d)c+=d; else break; } cout<<c<<endl; return 0; } |
English