#include <iostream>
#include <set>
using namespace std;
const int maxN=1e3;
int n, k, arr[maxN];
long long result=0;
set<pair<int, int>>S;
void flood_fill(int crr){
S.erase({arr[crr], crr});
for(int i=crr-1; i>=0; i--){
if(arr[i]-arr[i+1] > k)break;
S.erase({arr[i], i});
if(arr[i+1]-arr[i] > k){
int inc = arr[i+1]-k-arr[i];
arr[i] += inc;
result += inc;
}
}
for(int i=crr+1; i<n; i++){
if(arr[i]-arr[i-1] > k)break;
S.erase({arr[i], i});
if(arr[i-1]-arr[i] > k){
int inc = arr[i-1]-k-arr[i];
arr[i] += inc;
result += inc;
}
}
}
int main(){
ios_base::sync_with_stdio(false);
cin.tie(NULL);
cin>>n>>k;
for(int i=0; i<n; i++){
cin>>arr[i];
S.insert({arr[i], i});
}
while(!S.empty())flood_fill(S.rbegin()->second);
cout<<result;
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 | #include <iostream> #include <set> using namespace std; const int maxN=1e3; int n, k, arr[maxN]; long long result=0; set<pair<int, int>>S; void flood_fill(int crr){ S.erase({arr[crr], crr}); for(int i=crr-1; i>=0; i--){ if(arr[i]-arr[i+1] > k)break; S.erase({arr[i], i}); if(arr[i+1]-arr[i] > k){ int inc = arr[i+1]-k-arr[i]; arr[i] += inc; result += inc; } } for(int i=crr+1; i<n; i++){ if(arr[i]-arr[i-1] > k)break; S.erase({arr[i], i}); if(arr[i-1]-arr[i] > k){ int inc = arr[i-1]-k-arr[i]; arr[i] += inc; result += inc; } } } int main(){ ios_base::sync_with_stdio(false); cin.tie(NULL); cin>>n>>k; for(int i=0; i<n; i++){ cin>>arr[i]; S.insert({arr[i], i}); } while(!S.empty())flood_fill(S.rbegin()->second); cout<<result; return 0; } |
English