Submission #383730

#	Time	Username	Problem	Language	Result	Execution time	Memory
383730		shafinalam	Split the sequence (APIO14_sequence)	C++14	50 / 100	2090 ms	24684 KiB

sequence

#include <bits/stdc++.h>
 
using namespace std;
 
typedef long long ll;
typedef pair<ll,ll>pii;
typedef pair<pii,pii>piii;
const int mxn = 1e4+5;
#define inf 1e16
#define all(v) v.begin(), v.end()
#define  FAST  ios_base::sync_with_stdio(0);cout.tie(0)
 
 
ll arr[mxn];
ll dp[mxn][205];
ll pre[mxn];
int dir[mxn][205];
int n, k;
 
ll solve(int pos, int x)
{
    if(pos>n) {
        return -inf;
    }
    if(x==k) return 0;
    if(dp[pos][x]!=-1) return dp[pos][x];
    ll sum = 0, ret = -inf;
    for(int i = pos; i <= n; i++) {
        sum+=arr[i];
        ll tmp = max(ret, solve(i+1, x+1)+(sum*pre[i+1]));
        if(tmp>ret) {
            ret = tmp;
            dir[pos][x] = i+1;
        }
    }
    return dp[pos][x] = ret;
}
void pathprint(int pos, int x)
{
    if(x==k) return;
    cout << dir[pos][x]-1 << ' ';
    pathprint(dir[pos][x], x+1);
}
int main()
{
    FAST;    
    cin >> n >> k;
    for(int i = 1; i <= n; i++) {
        cin >> arr[i];
    }
    pre[n] = arr[n];
    for(int i = n-1; i >= 1; i--) {
        pre[i] = pre[i+1]+arr[i];
    }
    memset(dp, -1, sizeof dp);
    ll ans = solve(1, 0);
    cout << ans << '\n';
    pathprint(1, 0);
    return 0;
}

• Subtask 1: 11 / 11
• Subtask 2: 11 / 11
• Subtask 3: 11 / 11
• Subtask 4: 17 / 17
• Subtask 5: 0 / 21
• Subtask 6: 0 / 29