Submission #338306

#	Time	Username	Problem	Language	Result	Execution time	Memory
338306		Tosic	K blocks (IZhO14_blocks)	C++14	0 / 100	1 ms	508 KiB

blocks

#include <bits/stdc++.h>
#define maxn 100100
using namespace std;

int n, k;
int a[maxn], dp[maxn][110];

int main(){
	ios_base::sync_with_stdio(0);
	cout.tie(0);
	cin.tie(0);
	cin >> n >> k;
	for(int i = 1; i <= n; ++i){
		cin >> a[i];
	}
	dp[0][0] = 0;
	for(int i = 0; i < n; ++i){
		dp[i+1][1] = max(dp[i][1], a[i+1]);
	}
	for(int j = 2; j <= k; ++j){
		vector<pair<int, int> > dpStk;
		for(int i = j; i <= n; ++i){
			int tmpM = dp[i-1][j-1];
			while(!dpStk.empty() and dpStk.back().first < a[i]){
				if(dpStk.back().first + dpStk.back().second < tmpM+a[i]){
					tmpM = dpStk.back().second;
				}
				dpStk.pop_back();
			}
			if(dpStk.empty() or dpStk.back().first+dpStk.back().second >= tmpM+a[i]){
				dpStk.push_back({a[i], tmpM});
				//cerr << tmpM << '\n';
			}
			dp[i][j] = tmpM+a[i];
		}
	}
	cout << dp[n][k];
}

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