Submission #789496

#	Time	Username	Problem	Language	Result	Execution time	Memory
789496		Warinchai	Feast (NOI19_feast)	C++14	100 / 100	103 ms	8056 KiB

feast

#include<bits/stdc++.h>
using namespace std;
long long ar[300005];
long long sum[300005];
long long n,k;
pair<long long,int> solve(long long l){
    pair<long long,int>dp={0,0},best={0,0};
    for(int i=1;i<=n;i++){
        dp=max(dp,{sum[i]-l+best.first,best.second-1});
        best=max(best,{dp.first-sum[i],dp.second});
    }
    dp.second*=-1;
    return dp;
}
int main(){
    ios_base::sync_with_stdio(false);
    cin.tie(0);
    cin>>n>>k;
    long long sm=0;
    for(int i=1;i<=n;i++){
        cin>>ar[i];
        sm+=ar[i];
        sum[i]=sm;
    }
    long long st=0,en=3e14+1;
    long long ans;
    while(st<=en){
        long long m=(st+en)/2;
        pair<long long,int>tmp=solve(m);
        //cout<<m<<" "<<tmp.first<<" "<<tmp.second<<endl;
        if(tmp.second<=k){
            en=m-1;
            ans=tmp.first+m*k;
        }else{
            st=m+1;
        }
    }
    cout<<ans;
}

• 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