Submission #874976

#	Time	Username	Problem	Language	Result	Execution time	Memory
874976		NintsiChkhaidze	Split the sequence (APIO14_sequence)	C++17	0 / 100	217 ms	131072 KiB

sequence

#include <bits/stdc++.h>
#define ll long long
#define pb push_back
#define s second
#define f first
#define int ll
using namespace std;
 
const int N = 1e5 + 3;
 
int a[N],p[N],dp[N][203],lst[N][203];

signed main() {
    ios_base::sync_with_stdio(0),cin.tie(NULL),cout.tie(NULL);
 
	int n,k;
	cin>>n>>k;
	
	for (int i = 1; i <= n; i++){
		cin >> a[i];
		p[i] = p[i - 1] + a[i];
	}
	
	for (int r = 2; r <= n + 1; r++)
		for (int kk = 1; kk <= k; kk++)
			dp[r][kk] = -1;
	
	for (int i = 2; i <= n; i++){
		for (int j = 1; j <= k; j++){
			if (dp[i - 1][j] > dp[i][j]){
				dp[i][j] = dp[i - 1][j];
				lst[i][j] = lst[i - 1][j];
			}

			for (int l = 1; l < i; l++){
				if (dp[l - 1][j - 1] == -1) continue;

				int s1 = p[n] - p[i - 1],s2 = p[i - 1] - p[l - 1];
				int reward = s1*s2;
				int cur = dp[l][j - 1] + reward;
		
				if (cur > dp[i][j]){
					dp[i][j] = cur;
					lst[i][j] = l;
				}
			}
		}
	}
	cout<<dp[n][k]<<endl;
	
	vector <int> ans;
	int id = n;
	for (int i = k; i >= 1; i--){
		id = lst[id][i];
		ans.pb(id);
	}
	
	for (int i = k - 1; i >= 0; i--)
		cout<<ans[i]<<" ";
} 

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