Submission #48251

#	Time	Username	Problem	Language	Result	Execution time	Memory
48251		TheDarkning	Split the sequence (APIO14_sequence)	C++17	0 / 100	2056 ms	9576 KiB

sequence

/**
                  ▄█▀ ▀█▀ ▄▀▄ █▀ █▄█▄█ ▄▀▄ █▀ ▄█▀
                  <⇋⇋⇋⋛∰≓⊂(⌒,_ゝ⌒)⊃≓∰⋛⇋⇋⇋>

            ♔♕♖♗♘♙ ☜❷☞✪ ィℋ६ ≈ ᗫẵℜℵĬŊĞ ✪☜❷☞ ♚♛♜♝♞♟
            ♔♕♖♗♘♙                             ♚♛♜♝♞♟
                      ˙·٠•●♥ Ƹ̵̡Ӝ̵̨̄Ʒ ♥●•٠·˙
7 3
4 1 3 4 0 2 3

**/

#include <iostream>
#include <cstdio>
#include <vector>
#include <algorithm>
#include <iomanip>
#include <time.h>
#include <map>
#include <deque>
#include <assert.h>
#include <queue>
#include <string>
#include <memory.h>
#include <set>
#include <math.h>

#define sz(s) s.size()
#define pb emplace_back
#define fr first
#define sc second
#define mk make_pair
#define all(s) s.begin(), s.end()
#define ok puts("ok");

using namespace std;

const int N = 1e5 + 5;
const int inf = 1e18 + 7;

int n, k, ar[N], pref[N], pr[209][N], last = 1, cur = 1;

long long dp[2][N];

main(){
    cin >> n >> k;
    for( int i = 1; i <= n; i++ ){
        cin >> ar[i];
        pref[ i ] = pref[ i - 1 ] + ar[ i ];
    }
    k++;
    for( int i = 1; i <= k; i++ ){
            cur = 1;
            last = 1;
        for( int j = 1; j <= n; j++ ){
            for( int kk = last; kk <= j;kk++ ){
                if( dp[ i & 1 ][ j ] < dp[ (i & 1) ^ 1 ][ kk - 1 ] + ( pref[ j ] - pref[ kk - 1 ] ) * ( pref[ n ] - pref[j] )  ){
                    dp[ i & 1 ][ j ] = dp[ (i  & 1) ^ 1 ][ kk - 1 ] + (pref[j] - pref[kk - 1]) * (pref[n] - pref[j]) ;
                    cur = kk;
                    pr[ i ][ j ] = kk - 1;
                }
            }
            last = cur;
        }
        for (int j = 1; j <= n; j++)
            dp[(i  & 1) ^ 1][j] = 0;
    }
    cout << dp[ k & 1 ][ n ] << endl;
    while( k > 1 ){
        cout << pr[ k ][ n ] << ' ';
        n = pr[ k ][ n ];
        k--;
    }
}

Compilation message (stderr)

sequence.cpp:39:22: warning: overflow in implicit constant conversion [-Woverflow]
 const int inf = 1e18 + 7;
                 ~~~~~^~~
sequence.cpp:45:6: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
 main(){
      ^

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