oj.uz

Submission #964188

# Submission time Handle Problem Language Result Execution time Memory
964188 2024-04-16T11:54:03 Z dubabuba Split the sequence (APIO14_sequence) C++14
0 / 100
 4 ms 604 KB 
#include <bits/stdc++.h>
using namespace std;

template<typename T> void ono_max(T &MAX, T CMP) { if(MAX < CMP) MAX = CMP; }

#define int long long

const int mxk = 22;
const int mxn = 1010;
int dp[mxn][mxk], a[mxn], n;
int ls[mxn][mxk], p[mxn], k;

signed main() {
	cin >> n >> k;
	if(n > mxn) return 1;
	for(int i = 1; i <= n; i++) {
		cin >> a[i];
		p[i] = p[i - 1] + a[i];
	}

	for(int i = 0; i < mxn; i++)
	for(int j = 0; j < mxk; j++) {
		dp[i][j] = LONG_LONG_MAX;
		if(j == 0) {
			dp[i][j] = p[i] * p[i];
			ls[i][j] = i;
		}
	}

	dp[1][0] = a[1] * a[1];

	for(int cur = 2; cur <= n; cur++)
	for(int las = 1; las < cur; las++) {
		int las_sum = p[cur] - p[las];

		for(int j = 0; j <= k && dp[las][j] != -1; j++) {
			int CMP = dp[las][j] + las_sum * las_sum;
			if(dp[cur][j + 1] > CMP) {
				dp[cur][j + 1] = CMP;
				ls[cur][j + 1] = las;
			}
		}
	}

	int t = (p[n] * p[n] - dp[n][k]) / 2;
	cout << t << endl;

	vector<int> v;
	for(int i = k; i >= 0; i--) {
		// cout << n << ' ' << i << endl;
		if(n == 0) return i;
		v.push_back(n);
		n = ls[n][i];
	}

	while(v.size() > 1) {
		cout << v.back() << ' ';
		v.pop_back();
	}
	return 0;
}

Subtask #1
0 / 11.0

# Verdict Execution time Memory Grader output
1 Incorrect 1 ms 600 KB Unexpected end of file - int32 expected
2 Halted 0 ms 0 KB -

Subtask #2
0 / 11.0

# Verdict Execution time Memory Grader output
1 Incorrect 1 ms 604 KB Unexpected end of file - int32 expected
2 Halted 0 ms 0 KB -

Subtask #3
0 / 11.0

# Verdict Execution time Memory Grader output
1 Incorrect 1 ms 600 KB Unexpected end of file - int32 expected
2 Halted 0 ms 0 KB -

Subtask #4
0 / 17.0

# Verdict Execution time Memory Grader output
1 Incorrect 4 ms 600 KB Unexpected end of file - int32 expected
2 Halted 0 ms 0 KB -

Subtask #5
0 / 21.0

# Verdict Execution time Memory Grader output
1 Runtime error 0 ms 348 KB Execution failed because the return code was nonzero
2 Halted 0 ms 0 KB -

Subtask #6
0 / 29.0

# Verdict Execution time Memory Grader output
1 Runtime error 1 ms 348 KB Execution failed because the return code was nonzero
2 Halted 0 ms 0 KB -