Submission #671975

#	Time	Username	Problem	Language	Result	Execution time	Memory
671975		ReLice	K blocks (IZhO14_blocks)	C++14	100 / 100	168 ms	80636 KiB

blocks

#include<bits/stdc++.h>
using namespace std;
#define endl "\n"
#define ll long long
#define ld long double
#define int long long
#define pb push_back
#define sz size()
#define fr first
#define sc second
void start(){
    ios_base::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);
}
const ll N=1e5+1;
const ll inf=1e9+7;
ll dp[105][N];
main(){
    start();
    ll n,k,i,j,b,mx=-1;
    cin>>n>>k;
    ll a[n+5];
	for(i=1;i<=n;i++) cin>>a[i];
	for(i=1;i<=n;i++) dp[1][i]=max(dp[1][i-1],a[i]);
	for(i=2;i<=k;i++)
		for(j=1;j<=n;j++)
			dp[i][j]=inf;
	for(i=2;i<=k;i++){
		stack<pair<ll,ll>> st;
		for(j=i;j<=n;j++){
			ll tmp=dp[i-1][j-1];
			while(!st.empty() && (st.top().fr<=a[j])){
				tmp=min(tmp,st.top().sc);
				st.pop();
			}
			if(st.empty() || st.top().fr+st.top().sc>=tmp+a[j])
				st.push({a[j],tmp});
			dp[i][j]=min(tmp+a[j],st.top().fr+st.top().sc);
		}
	}
	cout<<dp[k][n];
}

Compilation message (stderr)

blocks.cpp:19:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
   19 | main(){
      | ^~~~
blocks.cpp: In function 'int main()':
blocks.cpp:21:16: warning: unused variable 'b' [-Wunused-variable]
   21 |     ll n,k,i,j,b,mx=-1;
      |                ^
blocks.cpp:21:18: warning: unused variable 'mx' [-Wunused-variable]
   21 |     ll n,k,i,j,b,mx=-1;
      |                  ^~

• Subtask 1: 14 / 14
• Subtask 2: 18 / 18
• Subtask 3: 21 / 21
• Subtask 4: 47 / 47