Submission #334175

#	Time	Username	Problem	Language	Result	Execution time	Memory
334175		vipghn2003	Bigger segments (IZhO19_segments)	C++14	100 / 100	298 ms	74732 KiB

segments

#include<bits/stdc++.h>
#define ll long long
#define fi first
#define se second
#define pii pair<int,int>
#define mp make_pair
using namespace std;

const int N=5e5+5;
int n;
ll a[N];
map<int,int>dp[N];

int main()
{
    ios_base::sync_with_stdio(false);
    cin.tie(nullptr);
    cin>>n;
    for(int i=1;i<=n;i++)
    {
        cin>>a[i];
        a[i]+=a[i-1];
    }
    dp[1][0]=0;
    for(int i=1;i<n;i++)
    {
        while(dp[i].size()>2) dp[i].erase(dp[i].begin());
        for(auto&x:dp[i])
        {
            dp[i+1][x.fi]=max(dp[i+1][x.fi],x.se);
            int v=lower_bound(a+1,a+n+1,2*a[i]-a[x.se])-a;
            if(v<=n) dp[v][x.fi+1]=max(dp[v][x.fi+1],i);
        }
    }
    while(dp[n].size()>1) dp[n].erase(dp[n].begin());
    auto res=*dp[n].begin();
    cout<<res.fi+1;
}

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