Submission #537228

#	Time	Username	Problem	Language	Result	Execution time	Memory
537228		LucaDantas	Split the sequence (APIO14_sequence)	C++17	60 / 100	110 ms	131072 KiB

sequence

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

constexpr int maxn = 2e5+10, maxk = 210;

int a[maxn], suf[maxn];

int itv(int l, int r) { return suf[l] - suf[r+1]; }

long long dp[maxn][maxk];
int opt[maxn][maxk];

void solve(int l, int r, int l2, int r2, int k) {
	if(l > r) return;

	int m = (l+r) >> 1;
	
	pair<long long, int> ans;
	for(int i = min(m-1, r2); i >= l2; i--)
		ans = max(ans, {dp[i][k-1] + 1ll * suf[m+1] * itv(i+1, m), i});

	dp[m][k] = ans.first;
	opt[m][k] = ans.second;

	solve(l, m-1, l2, ans.second, k);
	solve(m+1, r, ans.second, r2, k);
}

int main() {
	int n, k; scanf("%d %d", &n, &k);
	for(int i = 1; i <= n; i++)
		scanf("%d", a+i);
	
	for(int i = n; i >= 1; i--)
		suf[i] = suf[i+1] + a[i];

	for(int i = 1; i < n; i++)
		dp[i][1] = 1ll * suf[i+1] * itv(1, i);

	for(int i = 2; i <= k; i++)
		solve(1, n-1, 1, n-1, i);
	
	pair<long long, int> best;
	for(int i = 1; i <= n; i++)
		best = max(best, {dp[i][k], i});
	
	vector<int> ans;
	int i = best.second;
	for(int j = 0; j < k; j++)
		ans.push_back(i), i = opt[i][k-j];
	
	printf("%lld\n", best.first);
	reverse(ans.begin(), ans.end());
	for(int x : ans)
		printf("%d ", x);
	puts("");
}

Compilation message (stderr)

sequence.cpp: In function 'int main()':
sequence.cpp:30:17: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
   30 |  int n, k; scanf("%d %d", &n, &k);
      |            ~~~~~^~~~~~~~~~~~~~~~~
sequence.cpp:32:8: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
   32 |   scanf("%d", a+i);
      |   ~~~~~^~~~~~~~~~~

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