Submission #226253

#	Time	Username	Problem	Language	Result	Execution time	Memory
226253		bharat2002	Split the sequence (APIO14_sequence)	C++14	0 / 100	84 ms	131076 KiB

sequence

/*input
7 3
4 1 3 4 0 2 3
*/
#include<bits/stdc++.h>
using namespace std;
const int N=1e5 + 100;
const int mod=1e9 + 7;
#define int long long
const int inf=1e18;
#define pii pair<int, int>
#define f first
#define s second 
#define mp make_pair
#define FOR(i, n) for(int i=1;i<=n;i++)
#define TRACE(x) cerr << #x << " = " << x << endl 
//Trace prints the name of the variable and the value.
int dp[N][210], arr[N], n, k, pref[N], p[N][210];
stack<int> st;int ind;
double inter(int a, int b)
{
	double ret=(dp[a][ind] - dp[b][ind] + 0.0)/(pref[a] - pref[b] + 0.0);
	return ret;
}
void remove()
{
	while(st.size()>=3)
	{
		int i1=st.top(), i2, i3;st.pop();i2=st.top();st.pop();i3=st.top();
		st.pop();st.push(i3);
		if(inter(i1, i3)<inter(i2, i3)) st.push(i1);
		else {st.push(i2);st.push(i1);return;}
	}
}
signed main()
{
	ios_base::sync_with_stdio(false);cin.tie(NULL);cout.tie(NULL);
	cin>>n>>k;pref[0]=0;
	if(n==1)
	{
		cout<<0;return 0;
	}
	for(int i=1;i<=n;i++) {cin>>arr[i];pref[i]=arr[i] + pref[i-1];}
	for(int i=1;i<=n;i++)
	{
		for(int j=1;j<=k;j++) dp[i][j]=0;
	}
	for(int i=1;i<=n;i++)
	{
		dp[i][1]=pref[i]*(pref[n]-pref[i]);
	}
	for(int j=2;j<=k;j++)
	{
		st.push(j-1);ind=j-1;
		for(int i=j;i<=n;i++)
		{	
			while(st.size()>1)
			{
				int a=st.top();st.pop();int b=st.top();
				if(dp[a][ind] - pref[a]*(pref[n]-pref[i])<dp[b][ind]-pref[b]*(pref[n]-pref[i]))
				{
					st.push(b);
				}
				else {st.push(b);st.push(a);break;}
			}
			int opt=st.top();
			dp[i][j]=dp[opt][j-1] -pref[opt]*(pref[n]-pref[i]);
			dp[i][j]+=(pref[i])*(pref[n]-pref[i]);p[i][j]=opt;
			st.push(j);remove();
		}
		while(!st.empty()) st.pop();
	}
	int mi, mval=-inf;
	for(int j=1;j<=n;j++) 
	{
		if(mval<dp[j][k])
		{
			mi=j;mval=dp[j][k];
		}
	}
	cout<<mval<<endl;
	stack<int> st;st.push(mi);
	while(st.size()<k)
	{
		int cur=p[st.top()][k-st.size()+1];st.push(cur);
	}
	while(!st.empty()) {cout<<st.top()<<" ";st.pop();}
}

Compilation message (stderr)

sequence.cpp: In function 'int main()':
sequence.cpp:83:17: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
  while(st.size()<k)
        ~~~~~~~~~^~

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