Submission #1087458

#	Time	Username	Problem	Language	Result	Execution time	Memory
1087458		makrav	Bigger segments (IZhO19_segments)	C++14	0 / 100	1 ms	348 KiB

segments

#include <bits/stdc++.h>

using namespace std;
using ll = long long;

#define all(x) (x).begin(), (x).end()
#define sz(x) (int)(x).size()
#define pb push_back
#define ff first
#define sc second
#define int ll

void solve() {
    int n; cin >> n;
    vector<int> a(n);
    for (int i = 0; i < n; i++) cin >> a[i];
    int last = 0, cs = 0, ans = 0;
    vector<vector<int>> segs;
    int curl = 0;
    for (int i = 0; i < n; i++) {
        cs += a[i];
        if (cs >= last) {
            last = cs;
            segs.pb({curl, i, cs});
            curl = i + 1;
            ans++;
            cs = 0;
        }
    }
    if (cs == 0) {
        cout << ans << '\n';
        return;
    }
    vector<int> pref(n + 1);
    for (int i = 1; i <= n; i++) {
        pref[i] = pref[i - 1] + a[i - 1];
    }
    int LS = 0, LG = 0;
    for (int i = 1; i < sz(segs); i++) {
        int curm = segs[i][0], curr = segs[i][1];
        while (true) {
            if (curm > curr) break;
            if (pref[curr + 1] - pref[curm] >= pref[curm] - pref[LG] && pref[curm] - pref[LG] >= LS) {
                curm++;
            } else break;
        }
        curm--;
        LS = pref[curm] - pref[LG];
        //cout << LS << ' ' << LG << '\n';
        LG = curm;
    }
    //cout << LS << ' ' << LG << ' ' << ans << '\n';
    if (pref[n] - pref[segs.back()[1] + 1] >= pref[segs.back()[1] + 1] - pref[LG]) cout << ans + 1 << '\n';
    else cout << ans << '\n';
}

signed main() {
    int tt = 1;
    #ifdef LOCAL 
        freopen("in.txt", "r", stdin);
        freopen("out.txt", "w", stdout);
        cin >> tt;
    #else
        ios::sync_with_stdio(false); 
        cin.tie(0); cout.tie(0);
    #endif

    while (tt--) {
        solve();
    }

    return 0;
}

• Subtask 1: 0 / 13
• Subtask 2: 0 / 14
• Subtask 3: 0 / 10
• Subtask 4: 0 / 36
• Subtask 5: 0 / 27