Submission #552518

#	Time	Username	Problem	Language	Result	Execution time	Memory
552518		tht2005	Split the sequence (APIO14_sequence)	C++17	0 / 100	2099 ms	79288 KiB

sequence

#include <bits/stdc++.h>
 
using namespace std;
 
typedef long long LL;
 
const int N = 100001;
const int K = 201;
int n, kk, s[N], f[K][N];
LL a[N], b[N];
 
int main() {
    scanf("%d %d", &n, &kk);
    for(int i = 1; i <= n; ++i) {
        int a;
        scanf("%d", &a);
        s[i] = s[i - 1] + a;
    }
    memset(f, 0, sizeof(f));
    for(int t = 1; t <= kk; ++t) {
        for(int i = t + 1; i <= n; ++i) {
            for(int j = i - 1; j >= t; --j) {
                f[t][i] = max(f[t][i], f[t - 1][j] + s[j] * (s[i] - s[j]));
            }
        }
    }
    printf("%d\n", f[kk][n]);
    for(int i, j = n; kk--; ) {
        for(i = j; i--; ) {
            if(f[kk + 1][j] == f[kk][i] + s[i] * (s[j] - s[i])) {
                printf("%d ", i);
                j = i;
                break;
            }
        }
    }
    return 0;
}

Compilation message (stderr)

sequence.cpp: In function 'int main()':
sequence.cpp:13:10: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
   13 |     scanf("%d %d", &n, &kk);
      |     ~~~~~^~~~~~~~~~~~~~~~~~
sequence.cpp:16:14: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
   16 |         scanf("%d", &a);
      |         ~~~~~^~~~~~~~~~

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