Submission #647558

#	Time	Username	Problem	Language	Result	Execution time	Memory
647558		x0r	K blocks (IZhO14_blocks)	C++17	0 / 100	1 ms	340 KiB

blocks

#pragma GCC optimize ("O2")
#include <bits/stdc++.h>

#define ll long long
#define ld long double
#define fi first 
#define se second
#define pll pair < ll, ll >
#define pii pair < int, int >
#define fast ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);

using namespace std;

const string NAME = "";
const string NAME2 = "TEST";
const ld ESP = 1e-9;
const ll INF = 1e9 + 7;
const ll LINF = 1E18;
const ll nmax = 2e5;
const ll MOD = 1e9 + 7;
const ll base = 2309;

void fre() {
	string finp = NAME + ".inp";
	string fout = NAME + ".out";
	freopen(finp.c_str(), "r", stdin);
	freopen(fout.c_str(), "w", stdout);
}

int n, a[100003], st[100003][20], k, dp[103][100003];

int get(int l, int r) {
	int k = log2(r - l + 1);
	return max(st[l][k], st[r - (1 << k) + 1][k]);
}

void calc(int id, int l, int r, int u, int v) {
	if (l > r) return ;
	int mid = (l + r) / 2, pos = id - 1;
	for (int i = u; i <= min(r, mid - 1); i++) {
		//cout << mid << " " << i << " " << dp[id - 1][i] << " " << get(i + 1, mid) << '\n';
		if (dp[id][mid] > dp[id - 1][i] + get(i + 1, mid)) {
			dp[id][mid] = dp[id - 1][i] + get(i + 1, mid);
			pos = i;
		}
	}
	calc(id, l, mid - 1, u, pos);
	calc(id, mid + 1, r, pos, v);
}

int main() {
	fast;
	cin >> n >> k;
	for (int i = 1; i <= n; i++) {
		cin >> a[i];
		st[i][0] = a[i];
	}
	for (int j = 1; (1 << j) <= n; j++) 
		for (int i = 1; i + (1 << j) - 1 <= n; i++) {
			st[i][j] = max(st[i][j - 1], st[i + (1 << (j - 1))][j - 1]);
			//cout << i << " " << j <<  " " << st[i][j] << '\n';
		}
	for (int i = 2; i <= k; i++)
		for (int j = 1; j <= n; j++) dp[i][j] = INF;
	for (int i = 1; i <= n; i++) dp[1][i] = get(1, i);
	for (int i = 2; i <= k; i++) calc(i, 1, n, 1, n);
	cout << dp[k][n] << '\n';
}

Compilation message (stderr)

blocks.cpp: In function 'void fre()':
blocks.cpp:26:9: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   26 |  freopen(finp.c_str(), "r", stdin);
      |  ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~
blocks.cpp:27:9: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   27 |  freopen(fout.c_str(), "w", stdout);
      |  ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~

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