Submission #710624

#	Time	Username	Problem	Language	Result	Execution time	Memory
710624		PoonYaPat	Bigger segments (IZhO19_segments)	C++14	100 / 100	404 ms	44268 KiB

segments

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<ll,ll> pii;

int n;
ll qs[500001],dp[500001];
set<pii> s;

int main() {
    ios_base::sync_with_stdio(0); cin.tie(0);
    cin>>n;
    for (int i=1; i<=n; ++i) {
        cin>>qs[i];
        qs[i]+=qs[i-1];
    }
    s.insert(pii(0,0));

    for (int i=1; i<=n; ++i) {
        auto it=s.upper_bound(pii(qs[i],INT_MAX));
        --it;

        ll pos=(*it).second, nw=2*qs[i]-qs[pos];
        dp[i]=dp[pos]+1;

        it=s.upper_bound(pii(nw,INT_MAX));
        auto h=it; --h;
        if (h!=s.begin() && dp[(*h).second]>dp[i]) continue;

        it=s.upper_bound(pii(nw,-INT_MAX));
        while (it!=s.end()) {
            if ((dp[(*it).second]<dp[i]) || (dp[(*it).second]==dp[i] && (*it).second<i)) {
                auto k=it++;
                s.erase(k);
            } else break;
        }

        s.insert(pii(nw,i));
    }
    cout<<dp[n];
}

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