Submission #1149827

#	Time	Username	Problem	Language	Result	Execution time	Memory
1149827		dosts	Growing Vegetables is Fun 4 (JOI21_ho_t1)	C++20	100 / 100	15 ms	6620 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+2,0);
    for (int i=1;i<=n;i++) cin >> a[i];
    vi x(n+1,0),y(n+2,0);
    for (int i=1;i<=n;i++) {
        x[i] = x[i-1]+max(0ll,a[i-1]-a[i]+1);
    }
    for (int i=n;i>=1;i--) {
        y[i] = y[i+1]+max(0ll,a[i+1]-a[i]+1);
    }
    int ans = inf;
    vi mni(n+1);
    for (int i=1;i<=n;i++) {
        int lift = max({(a[i-1]+x[i-1])+1,(a[i+1]+y[i+1])+1,a[i]+max(x[i-1],y[i+1])})-(a[i]+max(x[i-1],y[i+1]));
        int sol = x[i-1],sag = y[i+1];
        int anss = max(sol,sag)+lift;
        ans = min(ans,anss);
    }
    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: 40 / 40
• Subtask 2: 60 / 60