Submission #996936

#	Time	Username	Problem	Language	Result	Execution time	Memory
996936		ttamx	Feast (NOI19_feast)	C++17	100 / 100	94 ms	12888 KiB

feast

#include<bits/stdc++.h>
 
using namespace std;
 
typedef long long ll;
typedef pair<ll,int> p2;
 
const int N=3e5+5;
 
int n,k;
ll a[N],qs[N];
p2 dp[N];
 
p2 calc(ll lambda){
    p2 mx(0,0);
    for(int i=1;i<=n;i++){
        dp[i]=max(dp[i-1],{mx.first+qs[i]-lambda,mx.second+1});
        mx=max(mx,{dp[i].first-qs[i],dp[i].second});
    }
    return dp[n];
}
 
int main(){
    cin.tie(nullptr)->sync_with_stdio(false);
    cin >> n >> k;
    for(int i=1;i<=n;i++)cin >> a[i];
    for(int i=1;i<=n;i++)qs[i]=qs[i-1]+a[i];
    ll l=0,r=3e14;
    while(l<r){
        ll m=(l+r+1)/2;
        if(calc(m).second>=k)l=m;
        else r=m-1;
    }
    cout << calc(l).first+l*k;
}

• Subtask 1: 4 / 4
• Subtask 2: 8 / 8
• Subtask 3: 18 / 18
• Subtask 4: 10 / 10
• Subtask 5: 11 / 11
• Subtask 6: 20 / 20
• Subtask 7: 29 / 29