Submission #976125

#	Time	Username	Problem	Language	Result	Execution time	Memory
976125		nnin	Split the sequence (APIO14_sequence)	C++14	0 / 100	15 ms	2908 KiB

sequence

#include<bits/stdc++.h>
using namespace std;

int n, k;
long long arr[100005];
long long sum[100005];
double a[100005];

int main() {
    ios_base::sync_with_stdio(false); cin.tie(NULL);
    cin>>n>>k;
    for(int i=1;i<=n;i++) {
        cin>>a[i];
        a[i] += a[i-1];
    }
    double part = a[n]/(k+1);
    vector<int> idx;
    idx.push_back(0);
    for(int i=1;i<=k;i++) {
        int it = lower_bound(a+1+idx[i-1], a+1+n, a[idx[i-1]]+part)-a;
        if(it==idx[i-1]+1 || a[it]-a[idx[i-1]]-part < part-a[it-1]+a[idx[i-1]]) {
            idx.push_back(it);
        } else {
            idx.push_back(it-1);
        }
        part = (a[n]-a[idx[i]])/(k+1-i);
    }
    idx.push_back(n);
    for(int i=1;i<=k+1;i++) {
        arr[i-1] = (long long)a[idx[i]]-(long long)a[idx[i-1]]; 
    }
    for(int i=k;i>=0;i--) {
        sum[i] = sum[i+1]+arr[i];
    }
    long long ans = 0;
    for(int i=0;i<=k;i++) {
        ans += arr[i]*sum[i+1];
    }
    cout<<ans<<'\n';
    for(int i=1;i<idx.size()-1;i++) cout<<idx[i]<<' ';
}

/*
7 3
4 1 3 4 0 2 3

*/

Compilation message (stderr)

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

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