Submission #157639

#	Time	Username	Problem	Language	Result	Execution time	Memory
157639		gemini_man	K blocks (IZhO14_blocks)	C++11	0 / 100	7 ms	4472 KiB

blocks

#include <bits/stdc++.h>
#define up(i,a,b) for(int (i) = (a);(i)<=(b);++(i))
#define down(i,b,a) for(int (i) = (b);i>=(a);--i)
#define debug(x) cerr << (x) << '\n';
#define bits(x,i) ((x >> i) & 1)
using namespace std;
const int N = 100005;
const int K = 105;
const int INF = 2e9;
int n,k;
int dp[N][K];
int a[N];
int lf[N];
int main(){
	ios_base::sync_with_stdio(0); cin.tie(0);
	cin >> n >> k;
	up(i,1,n) cin >> a[i];
	up(i,0,n) up(j,0,k) dp[i][j] = INF;
	int curMax = 0;
	up(i,1,n){
		curMax = max(curMax, a[i]);
		dp[i][1] = curMax;
	}
 
	int s = 0;
	vector<int> sk;
	up(i,1,n){
		while (!sk.empty()) {
			if (a[sk.back()] <= a[i]) sk.pop_back();
			else break;
		}
		if (!sk.empty()) lf[i] = sk.back() + 1;
		else lf[i] = 1;
		sk.push_back(i); 
	}
	for( int i = 2;i <= n;++i){
		for(int j = 2; j <= min(i,k) ;++j){
			dp[i][j] = min(dp[max(lf[i], j-1)][j-1] + a[i], dp[lf[i]-1][j]);
		}
	}
	cout << dp[n][k];
}

Compilation message (stderr)

blocks.cpp: In function 'int main()':
blocks.cpp:25:6: warning: unused variable 's' [-Wunused-variable]
  int s = 0;
      ^

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