Submission #879737

#	Time	Username	Problem	Language	Result	Execution time	Memory
879737		iskhakkutbilim	Split the sequence (APIO14_sequence)	C++17	0 / 100	2033 ms	17244 KiB

sequence

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

#define int long long
#define ff first
#define ss second
#define all(a) a.begin(), a.end()
const int mod = 1e17;
const int N = 1e5;

/*
 7 3
 4 1 3 4 0 2 3 
*/
signed main(){
	ios::sync_with_stdio(0);
	cin.tie(0); cout.tie(0);
	int n, k; cin >> n >> k;
	vector< vector<int> > dp(n + 1, vector<int>(k + 1, -mod));
	vector< vector<int> > dp1(n + 1, vector<int>(k + 1, mod));
	vector<int> a(n + 1, 0);
	for(int i = 1;i <= n; i++) cin >> a[i];
	vector<int> suff(n + 1, 0), pref(n + 1, 0);
	suff[n] = a[n];
	for(int i = 1;i <= n; i++) pref[i] = pref[i-1] + a[i];
	for(int i = n-1;i >= 1; i--) suff[i] = suff[i + 1] + a[i];
	dp1[0][0] = 0, dp[0][0] = 0;
	
	for(int i = 1;i <= n; i++){
		for(int sub = 0; sub <= k; sub++){
			dp[i][sub] = max(dp[i-1][sub], dp[i][sub]);
			if(sub == 0) continue;
			for(int j = i; j >= 1; j--){
				int pr = pref[i] - pref[j-1];
				dp[i][sub] = max(dp[i][sub], dp[j-1][sub-1] + (suff[i+1] * pr));
			}
		}
	}
	vector<int> path;
	int ans = dp[n][k];
	cout << dp[n][k] << '\n';
	for(int i = n;i >= 1; i--){
		if(dp[i][k] != ans){
			ans = dp[i][k];
			k--;
			path.push_back(i);
		}
	}
	reverse(all(path));
	for(auto x : path) cout << x << " ";
	
	return 0;
}

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