Submission #533662

#	Time	Username	Problem	Language	Result	Execution time	Memory
533662		rumen_m	Aliens (IOI16_aliens)	C++17	60 / 100	2077 ms	6188 KiB

aliens

#include <sstream>
#include <vector>
#include <algorithm>
#include <cstring>
#include <cstdlib>
#include <iostream>
#include <string>
#include <cassert>
#include <ctime>
#include <map>
#include <math.h>
#include <cstdio>
#include <set>
#include <deque>
#include <memory.h>
#include <queue>

#pragma comment(linker, "/STACK:64000000")

typedef long long ll;
using namespace std;

const ll INF = (ll)(1e18);
const int MAXN = 1 << 17;

ll sqr(ll x) {
	return x * x;
}

vector<pair<int, int> > a;
ll dp1[MAXN], dp2[MAXN];

ll cost(int l, int i) {
	if (l >= i) return INF;
	return sqr(a[i - 1].second - a[l].first + 1) - sqr(max(0LL, (l ? a[l - 1].second : -INF) - a[l].first + 1));
}

void rec(int l, int r, int optL, int optR) {
	if (l > r) return;
	int m = (l + r) >> 1;
	int opt = -1;
	for (int i = optL; i <= optR; i++) {
		if (dp2[m] > dp1[i] + cost(i, m)) {
			dp2[m] = dp1[i] + cost(i, m);
			opt = i;
		}
	}
	rec(l, m - 1, optL, opt);
	rec(m + 1, r, opt, optR);
}

ll take_photos(int n, int m, int k, vector<int> x, vector<int> y) {
	a.clear();
	for (int i = 0; i < n; i++) {
		a.push_back({ min(x[i], y[i]), max(x[i], y[i]) });
	}
	sort(a.begin(), a.end(), [](const pair<int, int> &a, const pair<int, int> &b) {
		if (a.first != b.first) return a.first < b.first;
		return a.second > b.second;
	});
	vector<pair<int, int> > aa;
	for (int i = 0; i < (int)a.size(); i++) {
		if (!aa.empty() && a[i].first >= aa.back().first && a[i].second <= aa.back().second) continue;
		aa.push_back(a[i]);
	}
	a = aa;
	n = a.size();

	memset(dp1, 63, sizeof(dp1));
	dp1[0] = 0;
	ll ans = INF;
	for (int i = 1; i <= k; i++) {
		memset(dp2, 63, sizeof(dp2));
		rec(1, n, 0, n - 1);
		ans = min(ans, dp2[n]);
		swap(dp1, dp2);
	}
	return ans;
}

// BEGIN CUT
/*
int main() {
#ifdef _MSC_VER
	freopen("input.txt", "r", stdin);
#endif
	int n, m, k;
	scanf("%d%d%d", &n, &m, &k);
	vector<int> x(n), y(n);
	for (int i = 0; i < n; i++) cin >> x[i] >> y[i];
	ll ans = take_photos(n, m, k, x, y);
	
    // BEGIN SECRET
    puts("098d134608c94f7413faac591054ee35");
    // END SECRET

	
	cout << ans << endl;
	return 0;
}
*/
// END CUT

Compilation message (stderr)

aliens.cpp:18: warning: ignoring '#pragma comment ' [-Wunknown-pragmas]
   18 | #pragma comment(linker, "/STACK:64000000")
      |

• Subtask 1: 4 / 4
• Subtask 2: 12 / 12
• Subtask 3: 9 / 9
• Subtask 4: 16 / 16
• Subtask 5: 19 / 19
• Subtask 6: 0 / 40