Submission #519563

#TimeUsernameProblemLanguageResultExecution timeMemory
519563Dasha_GnedkoBigger segments (IZhO19_segments)C++17
100 / 100
253 ms69780 KiB
#include <bits/stdc++.h>

//#include <ext/pb_ds/detail/standard_policies.hpp>
//#include <ext/pb_ds/assoc_container.hpp>
//#include <ext/pb_ds/tree_policy.hpp>

//#pragma GCC optimize("Ofast")
//#pragma GCC target("avx2")
//#pragma GCC target("sse,sse2,sse3,ssse3,sse4")

using namespace std;

//using namespace __gnu_pbds;
//template <typename T> using ordered_set = tree <T, null_type, less < T >, rb_tree_tag, tree_order_statistics_node_update>;

mt19937 gen(time(0));

#define ll long long
#define ld long double
#define pb push_back
#define F first
#define S second
#define TIME clock() * 1.0 / CLOCKS_PER_SEC
#define sz(a) int32_t(a.size())
#define endl '\n'
//#define int long long

const int N = 5e5 + 100;
const int M = 31;
const int mod = 1e9 + 7;
const ll inf = 1e18 + 7;

ll pref[N];
int a[N];
vector < int > add[N];
set < int > st;
pair < int, ll > dp[N];

ll get(int l, int r)
{
    if (l > r) return 0;
    ll ans = pref[r];
    if (l) ans -= pref[l - 1];
    return ans;
}

void upd(pair < int, ll > &a, pair < int, ll > b)
{
    if (a.F < b.F) a = b;
    else if (a.F == b.F) a.S = min(a.S, b.S);
}

int32_t main()
{
    ios_base::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);

#ifdef LOCAL
    freopen("input.txt", "r", stdin);
    freopen("output.txt", "w", stdout);
#else
//    freopen("input.txt", "r", stdin);
//    freopen("output.txt", "w", stdout);
#endif // LOCAL

    int n;
    cin >> n;
    for(int i = 0; i < n; i++)
    {
        cin >> a[i];
        if (i) pref[i] = pref[i - 1];
        pref[i] += a[i];
    }
    dp[0] = {1, a[0]};
    for(int i = 0; i < n; i++)
    {
        dp[i] = {1, pref[i]};
        if (i) dp[i] = {dp[i - 1].F, dp[i - 1].S + a[i]};
        for(auto to: add[i])
            st.insert(to);
        if (sz(st))
        {
            int j = *st.rbegin();
            int l = j, r = i - 1;
            while (l <= r)
            {
                int mid = (l + r) / 2;
                if (dp[j].S + get(j + 1, mid) <= get(mid + 1, i))
                {
                    upd(dp[i], {dp[j].F + 1, get(mid + 1, i)});
                    l = mid + 1;
                }
                else r = mid - 1;
            }
        }
        int l = i + 1, r = n - 1, gr = n;
        while (l <= r)
        {
            int mid = (l + r) / 2;
            ll cnt = pref[mid] - pref[i];
            if (cnt >= dp[i].S)
            {
                gr = mid;
                r = mid - 1;
            }
            else l = mid + 1;
        }
        add[gr].pb(i);
    }
    cout << dp[n - 1].F;
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...