Submission #1231824

#TimeUsernameProblemLanguageResultExecution timeMemory
123182412345678Feast (NOI19_feast)C++17
100 / 100
108 ms9824 KiB
#include <bits/stdc++.h>

using namespace std;

#define ll long long

const ll nx=3e5+5, inf=1e15;

ll n, k, a[nx], qs[nx], l=0, r=inf;

pair<ll, ll> solve(ll x)
{
    pair<ll, ll> mx, dp[n+1];
    for (int i=1; i<=n; i++)
    {
        dp[i]=max(dp[i-1], {mx.first+qs[i]-x, mx.second+1});
        mx=max(mx, {dp[i].first-qs[i], dp[i].second});
    }
    return dp[n];
}

int main()
{
    cin.tie(NULL)->sync_with_stdio(false);
    cin>>n>>k;
    for (int i=1; i<=n; i++) cin>>a[i], qs[i]=qs[i-1]+a[i];
    while (l<r)
    {
        ll md=(l+r+1)/2;
        if (solve(md).second>=k) l=md;
        else r=md-1;
    }
    //cout<<"debug "<<l<<' '<<solve(l).first<<' '<<solve(l).second<<'\n';
    cout<<solve(l).first+k*l;
}
#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...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...