#include<bits/stdc++.h>
using namespace std;
#define int long long
const int inf = 1e18;
signed main(){
int n; cin>>n;
vector<int> v(n); for(int& i : v) cin>>i;
vector<int> dp(n+1,inf), dp1(n+1,inf), mx(n+1,0), mx1(n+1,0);
dp[0] = 0;
mx[0] = 0;
vector<int> last(n+1,0), last1(n+1,0); // just add max(between act and last)
for(int i = 1; i<=n; i++){
// cerr<<i<<" "<<v[i-1]<<" "<<dp[i-1]<<" "<<mx[i-1]<<endl;
if(v[i-1] > mx[i-1]){
mx[i] = v[i-1];
dp[i] = dp[i-1]; continue;
}
int obj = mx[i-1] + 1;
mx[i] = obj;
dp[i] = dp[i-1] + abs(obj - v[i-1]);
}
dp1[n] = 0;
mx1[n] = 0;
for(int i = n-1; i>=0; i--){
// cerr<<"dp1 "<<i<<" "<<v[i]<<" "<<dp1[i+1]<<" "<<mx1[i+1]<<" ";
if(v[i] > mx1[i+1]){
// cerr<<"not"<<endl;
mx1[i] = v[i];
dp1[i] = dp1[i+1];
continue;
}
// cerr<<"yes ";
int obj = mx1[i+1] + 1;
// cerr<<obj<<" "<<v[i]<<endl;
dp1[i] = dp1[i+1] + obj - v[i];
mx1[i] = obj;
}
// mx1[n] = v[n-1]; mx[0] = v[0];
for(int i : mx1) cout<<i<<" "; cout<<endl;
int ans = min(dp[n],dp1[0]);
for(int i = 1; i<n; i++){
// cerr<<"HERE "<<i<<" "<<dp[i]<<" "<<dp1[i]<<" "<<mx1[i]<<" "<<mx[i]<<endl;
ans = min(ans, max(dp[i],dp1[i]) + max(0ll,mx1[i] + 1 - mx[i]));
}
cout<<ans<<endl;
}
Compilation message
Main.cpp: In function 'int main()':
Main.cpp:39:2: warning: this 'for' clause does not guard... [-Wmisleading-indentation]
39 | for(int i : mx1) cout<<i<<" "; cout<<endl;
| ^~~
Main.cpp:39:33: note: ...this statement, but the latter is misleadingly indented as if it were guarded by the 'for'
39 | for(int i : mx1) cout<<i<<" "; cout<<endl;
| ^~~~
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Incorrect |
0 ms |
344 KB |
Output isn't correct |
| 2 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Incorrect |
0 ms |
344 KB |
Output isn't correct |
| 2 |
Halted |
0 ms |
0 KB |
- |
