Submission #562808

#	Time	Username	Problem	Language	Result	Execution time	Memory
562808		Ooops_sorry	Split the sequence (APIO14_sequence)	C++14	50 / 100	2067 ms	24788 KiB

sequence

#include<bits/stdc++.h>

using namespace std;

#define pb push_back
#define ld long double
#define ll long long

mt19937 rnd(51);

const ll INF = 1e18;

signed main() {
#ifdef LOCAL
    freopen("input.txt", "r", stdin);
#endif // LCOAL
    ios_base::sync_with_stdio(0);
    cin.tie(0);
    int n, k;
    cin >> n >> k;
    vector<ll> a(n), pr(n);
    vector<vector<ll>> dp(n + 1, vector<ll>(k + 1, -INF));
    vector<vector<int>> par(n + 1, vector<int>(k + 1));
    for (int i = 0; i < n; i++) {
        cin >> a[i];
    }
    reverse(a.begin(), a.end());
    for (int i = 0; i < n; i++) {
        pr[i] = a[i];
        if (i > 0) {
            pr[i] += pr[i - 1];
        }
    }
    for (int i = 0; i < n; i++) {
        dp[i][0] = 0;
        for (int f = 0; f < k; f++) {
            ll sum = 0;
            for (int j = i + 1; j < n; j++) {
                sum += a[j];
                if (dp[i][f] + sum * pr[i] > dp[j][f + 1]) {
                    dp[j][f + 1] = dp[i][f] + sum * pr[i];
                    par[j][f + 1] = i;
                }
            }
        }
    }
    cout << dp[n - 1][k] << endl;
    vector<int> res;
    int i = n - 1, j = k, sum = 0;
    while (j > 0) {
        res.pb(i - par[i][j]);
        i = par[i][j];
        j--;
        sum += res.back();
    }
    int now = 0;
    for (auto to : res) {
        cout << (now += to) << ' ';
    }
    cout << endl;
    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