Submission #895878

#	Time	Username	Problem	Language	Result	Execution time	Memory
895878		thunopro	Split the sequence (APIO14_sequence)	C++14	0 / 100	29 ms	131072 KiB

sequence

#include<bits/stdc++.h>
using namespace std ; 
#define ll long long 
#define maxn 100009 
#define fi first 
#define se second 
#define pb push_back 
#define left id<<1 
#define right id<<1|1 
#define re exit(0); 
#define _lower(x) lower_bound(v.begin(),v.end(),x)-v.begin()+1  

const int mod = 1e9+7 ; 
const int INF = 1e9 ; 
const int LOG = 18 ; 

typedef vector<int> vi ; 
typedef pair<int,int> pii ; 
typedef vector<ll> vl ; 
typedef vector<pii> vii ; 
 
void add ( int &a , int b ) 
{
	a += b ;
	if ( a < 0 ) a += mod ; 
	if ( a >= mod ) a -= mod ; 
}

template < typename T > void chkmin ( T &a , T b ) { if ( a > b ) a = b ; } 
template < typename T > void chkmax ( T &a , T b ) { if ( a < b ) a = b ; } 

void rf () 
{
	freopen ("bai1.inp","r",stdin) ; 
//	freopen ("bai1.out","w",stdout) ; 
}

int _pow ( int a , int n ) 
{
	if ( n == 0 ) return 1 ; 
	int res = _pow (a,n/2) ; 
	if ( n % 2 ) return 1ll*res*res%mod*a%mod ; 
	else return 1ll*res*res%mod ; 
}

int n , k ; 
int a [maxn] ; 
int sum [maxn] ; 

ll dp [maxn][209] ; 
void trau () 
{
	memset ( dp , - 0x3f , sizeof dp ) ; 
	dp [0][0] = 0 ; 
	for ( int i = 1 ; i <= n ; i ++ ) 
	{
		for ( int j = 1 ; j <= i ; j ++ ) 
		{
			for ( int t = 1 ; t <= k ; t ++ )
			{
				chkmax (dp[i][t],dp[j-1][t-1]+1ll*(sum[i]-sum[j-1])*(sum[n]-sum[i])) ; 
			}
		}	
	}
	cout << dp [n][k] << "\n" ; 
	vi res ;
	int last = n ;
	for ( int i = n ; i >= 1 ; i -- ) 
	{
		if ( 1ll*(sum[last]-sum[i-1])*(sum[n]-sum[last]) + dp [i-1][k-1] == dp [last][k] && k > 1 ) 
		{
			res . pb (i-1) ; 
			last = i-1 ; 
			k -- ; 
		}
	}
	
	reverse(res.begin(),res.end()) ; 
	for ( auto x : res ) cout << x << " " ; 
}
int main () 
{
	ios_base::sync_with_stdio(0);
	cin.tie(0);cout.tie(0) ; 
//	rf () ; 
	cin >> n >> k ; k ++ ; 
	for ( int i = 1 ; i <= n ; i ++ ) cin >> a [i] , sum [i] = sum [i-1] + a [i] ;
	trau () ; 
}

// range , error , check speical , ... 
// find key , find key 
//'-std=c++11' stay hard 
// there is no tomorrow 

Compilation message (stderr)

sequence.cpp: In function 'void rf()':
sequence.cpp:34:10: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   34 |  freopen ("bai1.inp","r",stdin) ;
      |  ~~~~~~~~^~~~~~~~~~~~~~~~~~~~~~

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