Submission #395790

#	Time	Username	Problem	Language	Result	Execution time	Memory
395790		kaplanbar	Split the sequence (APIO14_sequence)	C++14	0 / 100	9 ms	2084 KiB

sequence

#include <bits/stdc++.h>
using namespace std;
using ll = long long;
int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(0);


    int n, k;
    cin >> n >> k;

    vector<ll> v(n);
    for(int i = 0; i < n; i++) cin >> v[i];

    vector<ll> pre(n);
    for(int i = 0; i < n; i++) {
        pre[i] = v[i];
        if(i) pre[i] += pre[i - 1];
    }

    auto get_sum = [&](int l, int r) -> ll {
        ll ret = pre[r];
        if(l) ret -= pre[l - 1];
        return ret;
    };

    ll s = get_sum(0, n - 1);

    ll ans = 0, cur = 0;
    vector<int> a;
    for(int i = 0; i < n; i++) {
        cur += v[i];
        if(cur >= s / (k + 1)) {
            ans += cur * (s - cur);
            a.push_back(i + 1);
            cur = 0;
        }
    }

    if(a.size() == k + 1) {
        a.pop_back();
    }
    else {
        ans += cur * (s - cur);
    }

    cout << ans / 2 << "\n";
    for(int x : a) cout << x << " ";

    return 0;
}

Compilation message (stderr)

sequence.cpp: In function 'int main()':
sequence.cpp:40:17: warning: comparison of integer expressions of different signedness: 'std::vector<int>::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
   40 |     if(a.size() == k + 1) {
      |        ~~~~~~~~~^~~~~~~~

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