Submission #1267952

#TimeUsernameProblemLanguageResultExecution timeMemory
1267952nguyenvietanhSplit the sequence (APIO14_sequence)C++20
71 / 100
80 ms165448 KiB
#include<bits/stdc++.h>

using namespace std;

#define int long long
#define fi first
#define se second
#define pii pair<int,int>
#define inp(name) freopen(name, "r", stdin);
#define out(name) freopen(name, "w", stdout);

const int N = 1e5 + 5;
int n, k;
int a[N];
int dp[205][N], prefix[N], pre[N][205];
int calc(int i, int j){
    if (i > j) return 0;
    return (prefix[j] - prefix[i - 1]) * (prefix[j] - prefix[i - 1]);
}
void solve(int l, int r, int pos, int lf, int rt){
    if (l > r) return;
    int mid = (l + r)/2;
    pair <int, int> min_dp = {LLONG_MAX, 0};
    for (int i = min(mid, rt); i >= lf; i --){
        if (dp[pos - 1][i - 1] + calc(i, mid) < min_dp.fi) min_dp = {dp[pos - 1][i - 1] + calc(i, mid), i};
    }
    dp[pos][mid] = min_dp.fi;
    pre[mid][pos] = min_dp.se - 1;
    solve(l, mid - 1, pos, lf, min_dp.se);
    solve(mid + 1, r, pos, min_dp.se, rt);
}
signed main() {
    ios_base::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);
    cin >> n >> k;
    int sum = 0;
    for (int i = 1; i <= n; i ++){
        cin >> a[i];
        prefix[i] = prefix[i - 1] + a[i];
    }
    for (int i = 1; i <= n; i ++) dp[0][i] = calc(1, i);
    for (int i = 1; i <= k; i ++){
        solve(1, n, i, 1, n);
    }
    cout << (prefix[n] * prefix[n] - dp[k][n])/2 << '\n';
    vector <int> ans;
    for (int i = k; i >= 1; i --){
        ans.push_back(pre[n][i]);
        n = pre[n][i];
    }
    while (ans.size()) cout << ans.back() << ' ', ans.pop_back();
}
#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...