Submission #964196

#	Time	Username	Problem	Language	Result	Execution time	Memory
964196		dubabuba	수열 (APIO14_sequence)	C++14	11 / 100	100 ms	772 KiB

sequence

#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;
const int inf = 2e18;
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 <= n; i++)
	for(int j = 0; j <= k; j++) {
		dp[i][j] = inf;
		if(j == 0) {
			dp[i][j] = p[i] * p[i];
			ls[i][j] = i;
		}
	}

	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] != inf; 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: 11 / 11
• Subtask 2: 0 / 11
• Subtask 3: 0 / 11
• Subtask 4: 0 / 17
• Subtask 5: 0 / 21
• Subtask 6: 0 / 29