Submission #1149822

#TimeUsernameProblemLanguageResultExecution timeMemory
1149822dostsGrowing Vegetables is Fun 4 (JOI21_ho_t1)C++20
0 / 100
1 ms328 KiB
#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+2,0);
    for (int i=1;i<=n;i++) cin >> a[i];
    vi b(n+1,0),c(n+2,0);
    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 l = a[i];
        int r = inf;
        while (l<=r) {
            int v = (l+r) >> 1;
            if ((a[i+1]+y[i+1]) < v && (a[i-1]+x[i-1]) < v && (v-a[i]) >= max(x[i-1],y[i+1])) r = v-1;
            else l = v+1;
        }
        ans = min(ans,l-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();
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...