#include <bits/stdc++.h>
using namespace std;
#define LL long long
#define all(a) a.begin(),a.end()
#define v vector
#define pb push_back
int main(){ios_base::sync_with_stdio(0);cin.tie(0);
int n, k;
LL wyn=0;
cin >> n >> k;
v<int>lista(n);
for(int i=0;i<n;i++) cin >> lista[i];
for(int i=0;i<n;i++){
int j=i-1, t=1;
while(t){
if(j<0) t=0;
else if(lista[j]+k>=lista[j+1]) t=0;
else{
wyn+=lista[j+1]-lista[j]-k;
lista[j]=lista[j+1]-k;
}
j--;
}
j=i+1;
t=1;
while(t){
if(j>=n) t=0;
else if(lista[j-1]<=lista[j]+k) t=0;
else{
wyn+=lista[j-1]-lista[j]-k;
lista[j]=lista[j-1]-k;
}
j++;
}
}
cout << wyn;
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 | #include <bits/stdc++.h> using namespace std; #define LL long long #define all(a) a.begin(),a.end() #define v vector #define pb push_back int main(){ios_base::sync_with_stdio(0);cin.tie(0); int n, k; LL wyn=0; cin >> n >> k; v<int>lista(n); for(int i=0;i<n;i++) cin >> lista[i]; for(int i=0;i<n;i++){ int j=i-1, t=1; while(t){ if(j<0) t=0; else if(lista[j]+k>=lista[j+1]) t=0; else{ wyn+=lista[j+1]-lista[j]-k; lista[j]=lista[j+1]-k; } j--; } j=i+1; t=1; while(t){ if(j>=n) t=0; else if(lista[j-1]<=lista[j]+k) t=0; else{ wyn+=lista[j-1]-lista[j]-k; lista[j]=lista[j-1]-k; } j++; } } cout << wyn; return 0; } |
English