Submission #1149804

#	Time	Username	Problem	Language	Result	Execution time	Memory
1149804		dosts	Growing Vegetables is Fun 4 (JOI21_ho_t1)	C++20	0 / 100	0 ms	324 KiB

Main

#include <bits/stdc++.h>
#pragma GCC target("avx2")
#pragma GCC optimize("O3,unroll-loops")
using namespace std;
#define int long long
#define pii pair<int,int>
#define ff first
#define ss second
#define sp << " " <<    
#define all(cont) cont.begin(),cont.end()
#define vi vector<int>

const int inf = 1e17,N = 3e5+1,MOD = 1e9+7,BL = 1000;

void solve() { 
    int n;
    cin >> n;
    vi a(n+1);
    for (int i=1;i<=n;i++) cin >> a[i];
    vi b(n+1),c(n+1);
    vi x(n+1,0),y(n+2,0);
    for (int i=1;i<=n;i++) {
        b[i] = max(a[i],b[i-1]+1);
        x[i] = x[i-1]+(b[i]-a[i]);
    }
    for (int i=n;i>=1;i--) {
        c[i] = max(a[i],c[i+1]+1);
        y[i] = y[i+1]+(c[i]-a[i]);
    }
    int ans = inf;
    for (int i=1;i<=n;i++) {
        int both = min(x[i-1],y[i+1]);
        int rem = max(x[i-1],y[i+1])-both;
        ans = min(ans,both+max(rem,max({a[i],b[i-1],c[i+1]})-a[i]));      
    }
    cout << ans << '\n';
}

int32_t main() { 
    ios_base::sync_with_stdio(0);cin.tie(0);cout.tie(0);
    #ifdef Dodi 
        freopen("in.txt","r",stdin);
        freopen("out.txt","w",stdout);
    #endif
    int t = 1;
    //cin >> t;
    while (t --> 0) solve();
}

• Subtask 1: 0 / 40
• Subtask 2: 0 / 60