Submission #1293477

#TimeUsernameProblemLanguageResultExecution timeMemory
1293477hahaFeast (NOI19_feast)C++20
22 / 100
71 ms4820 KiB
#include<bits/stdc++.h>
#define f first
#define s second
#define ll long long
using namespace std;
const int maxn=3e5+5;

int n,k;
int a[maxn];
ll sum[maxn],dp[305][305],Max[maxn];

void sub3()
{
    ll ans=0;
    ll Min=0;
    for(int i=1;i<=n;i++){
        ans=max(ans,sum[i]-Min);
        Min=min(Min,sum[i]);
    }
    cout<<ans;
}

void sub5()
{
    for(int i=0;i<=n;i++){
        for(int j=1;j<=k;j++) dp[i][j]=-1e18;
    }
    for(int i=1;i<=n;i++){
        for(int j=1;j<=k;j++){
            dp[i][j]=max(dp[i][j],dp[i-1][j]);
            for(int pre=1;pre<=i;pre++){
                dp[i][j]=max(dp[i][j],dp[pre-1][j-1]+sum[i]-sum[pre-1]);
                // dp[pre-1][j-1]-sum[pre-1] + sum[i]
            }
        }
    }
    ll ans=0;
    for(int i=1;i<=n;i++){
        ans=max(ans,dp[i][k]);
    }
    cout<<ans;
}

/*
void sub6()
{
    Max[0]=0;
    for(int i=1;i<=n;i++) Max[i]=max(Max[i-1],-sum[i]);
    for(int i=1;i<=n;i++) dp[i]=-1e18;
    ll ans=0;
    for(int j=1;j<=k;j++){
        for(int i=1;i<=n;i++){
            dp[i]=max(dp[i],dp[i-1]);
            dp[i]=max(dp[i],Max[i-1]+sum[i]);
            if(j==k) ans=max(ans,dp[i]);
        }
        Max[0]=-1e18;
        for(int i=1;i<=n;i++) Max[i]=max(Max[i-1],dp[i]-sum[i]);
        for(int i=0;i<=n;i++) dp[i]=-1e18;
    }
    cout<<ans;
}
*/
int main()
{
    ios_base::sync_with_stdio(0);
    cin.tie(0);cout.tie(0);
    cin>>n>>k;
    int neg=0,atv=0,zero=0;
    for(int i=1;i<=n;i++){
        cin>>a[i];
        if(a[i]<0) neg++;
        if(a[i]>0) atv++;
        if(a[i]==0) zero++;
        sum[i]=sum[i-1]+a[i];
    }
    if(atv+zero==n){
        cout<<sum[n];
        return 0;
    }
    else if(neg+zero==n){
        cout<<0;
        return 0;
    }
    else if(k==1) sub3();
    else sub5();
}
#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...