Submission #1002974

#	Time	Username	Problem	Language	Result	Execution time	Memory
1002974		pedroslrey	Izvanzemaljci (COI21_izvanzemaljci)	C++17	5 / 100	274 ms	25920 KiB

izvanzemaljci

#include <bits/stdc++.h>
#define int long long 

using namespace std;
using lli = long long;

pair<vector<int>, lli> calc(vector<pair<int, int>> points) {
	map<int, vector<pair<int, int>>> batch;
	for (auto [x, y]: points)
		batch[x].emplace_back(x, y);

	multiset<int> xs, ys;
	for (auto [x, y]: points) {
		xs.insert(x);
		ys.insert(y);
	}

	int d = max(*xs.rbegin() - *xs.begin(), *ys.rbegin() - *ys.begin());
	d = max(d, 1LL);

	int mx_x = -1e9, mn_x = 1e9, mx_y = -1e9, mn_y = 1e9;
	vector<int> ans{*xs.begin(), *ys.begin(), d, (int)2e9, (int)2e9, 1LL}; lli best = 1LL*d*d + 1;
	for (auto [k, abas]: batch) {
		for (auto [x, y]: abas) {
			assert(xs.find(x) != xs.end());
			assert(ys.find(y) != ys.end());

			if (xs.empty()) break;
			mx_x = max(mx_x, x); mn_x = min(mn_x, x);
			mx_y = max(mx_y, y); mn_y = min(mn_y, y);
			xs.erase(xs.find(x)); ys.erase(ys.find(y));
		}
		if (xs.empty()) break;

		int d1 = max(mx_x - mn_x, mx_y - mn_y);
		int d2 = max(*xs.rbegin() - *xs.begin(), *ys.rbegin() - *ys.begin());
		d1 = max(d1, 1LL);
		d2 = max(d2, 1LL);

		lli cost1 = 1LL*d1*d1;
		lli cost2 = 1LL*d2*d2;

		if (cost1 + cost2 < best) {
			best = cost1 + cost2;
			if (mn_x + d1 > k) 
				ans = vector<int>{k - d1, mn_y, d1, *xs.begin(), *ys.begin(), d2};
			else 
				ans = vector<int>{mn_x, mn_y, d1, *xs.begin(), *ys.begin(), d2};
		}
	}

	return {ans, best};
}

main() {
	int n, k;
	cin >> n >> k;

	vector<pair<int, int>> points(n);
	for (auto &[x, y]: points)
		cin >> x >> y;

	if (k == 1) {
		multiset<int> xs, ys;
		for (auto [x, y]: points) {
			xs.insert(x);
			ys.insert(y);
		}

		int d2 = max(*xs.rbegin() - *xs.begin(), *ys.rbegin() - *ys.begin());
		d2 = max(d2, 1LL);

		cout << *xs.begin() << " " << *ys.begin() << " " << d2 << "\n";
		return 0;
	}

	if (n == 1) {
		cout << points[0].first << " " << points[0].second << " 1\n";
		cout << points[0].first + 2 << " " << points[0].second + 2 << " 1\n";
		return 0;
	}

	auto [ans1, best1] = calc(points);
	for (auto &[x, y]: points) swap(x, y);
	auto [ans2, best2] = calc(points);

	if (best2 < best1) {
		swap(ans2, ans1);
		swap(ans1[0], ans1[1]);
		swap(ans1[3], ans1[4]);
	}

	for (int i = 0; i < 6; ++i)
		cout << ans1[i] << (i == 2 ? "\n" : " ");
	cout << "\n";
}

Compilation message (stderr)

izvanzemaljci.cpp:55:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
   55 | main() {
      | ^~~~

• Subtask 1: 5 / 5
• Subtask 2: 0 / 21
• Subtask 3: 0 / 12
• Subtask 4: 0 / 30
• Subtask 5: 0 / 32