Submission #333274

#	Time	Username	Problem	Language	Result	Execution time	Memory
333274		blue	Split the sequence (APIO14_sequence)	C++11	0 / 100	2087 ms	5396 KiB

sequence

#include <iostream>
using namespace std;

int main()
{
    int n, k;
    cin >> n >> k;

    int a[n+1];
    a[0] = 0;
    for(int i = 1; i <= n; i++)
    {
        cin >> a[i];
        a[i] += a[i-1];
    }

    int dp[n+1][k+1];
    int prev[n+1][k+1];
    for(int i = 1; i <= n; i++)
    {
        dp[i][0] = 0;
        prev[i][0] = 0;
        for(int l = 1; l < n && l <= k; l++)
        {
            dp[i][l] = 0;
            prev[i][l] = 0;

            for(int j = l; j < i; j++)
            {
                if(dp[j][l-1] + a[j] * (a[i] - a[j]) > dp[i][l])
                {
                    dp[i][l] = dp[j][l-1] + a[j] * (a[i] - a[j]);
                    prev[i][l] = j;
                }
            }
        }
    }
    cout << dp[n][k] << '\n';

    int temp = n;
    for(int l = k; l >= 1; l--)
    {
        cout << ((prev[temp][l] == 0) ? 1 : prev[temp][l]) << ' ';
        temp = prev[temp][l];
    }
    cout << '\n';
}

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