Submission #989417

#	Time	Username	Problem	Language	Result	Execution time	Memory
989417		cnn008	Split the sequence (APIO14_sequence)	C++17	100 / 100	967 ms	81744 KiB

sequence

#include "bits/stdc++.h"
using namespace std;
 
#ifdef N_N_C
#include "debug.h"
#else
#define cebug(...) "Arya"
#endif
 
#define ll long long
 
const int N=1e5+5;
const int mod=1e9+7;
 
int n,k,a[N],op[205][N];
ll dp[2][N],p[N];
void dnc(int j, int l, int r, int optl, int optr){
	if(l>r) return;
	int mid=(l+r)>>1;
	pair <ll,int> opt={0,0};
	for(int x=min(mid,optr); x>=optl; x--) opt=max(opt,{dp[(j&1)^1][x-1]+(p[mid]-p[x-1])*(p[n]-p[mid]),x});
	dp[j&1][mid]=opt.first;
	op[j][mid]=opt.second;
	dnc(j,l,mid-1,optl,opt.second);
	dnc(j,mid+1,r,opt.second,optr);
}
vector <int> trc;
void trace(int d, int j){
	if(!d) return;
	assert(j>0);
	trc.push_back(j);
	trace(d-1,op[d][j]-1);
}
void sol(){
	cin>>n>>k;
	for(int i=1; i<=n; i++) cin>>a[i],p[i]=p[i-1]+a[i];
	for(int j=1; j<=k; j++){
		for(int i=1; i<=n; i++) dp[j&1][i]=0;
		dnc(j,j,n-1,1,n);
	}
	ll ans=0;
	for(int i=1; i<n; i++) ans=max(ans,dp[k&1][i]);
	cout<<ans<<"\n";
	int pos=0;
	for(int i=1; i<n; i++) if(dp[k&1][i]==ans) pos=i;
	trace(k,pos);
	assert(!trc.empty());
	reverse(trc.begin(),trc.end());
	for(auto x:trc) cout<<x<<" ";
}
signed main(){
	ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    // freopen("sequence.in", "r", stdin);
    // freopen("sequence.out", "w", stdout);
    int tt=1;
    //cin>>tt; 
    while(tt--){
    	sol();
    }
    cerr << "\nTime elapsed: " << 1000.0 * clock() / CLOCKS_PER_SEC << " ms.\n";
    return 0;
}

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