Submission #983360

#	Time	Username	Problem	Language	Result	Execution time	Memory
983360		vjudge1	Circle selection (APIO18_circle_selection)	C++17	0 / 100	1018 ms	68356 KiB

circle_selection

#include <bits/stdc++.h>
#define int long long
using namespace std;
int n, i, j, k, x[300001], y[300001], r[300001], ans[300001];
bool u[300001];
multiset<pair<int, int>> ms, ms2, ms3, mss;

long double dist(long double x1, long double y1, long double x2, long double y2) {
	return (x2 - x1) * (x2 - x1) + (y2 - y1) * (y2 - y1);
}

bool inter(int i, int j) {
	return (r[i] + r[j]) * (r[i] + r[j]) >= dist(x[i], y[i], x[j], y[j]);
}

signed main() {
	cin.tie(0)->sync_with_stdio(0);
	#ifdef LOCAL
	freopen("input.txt", "r", stdin);
	#endif
	cin >> n;
	bool subtask2 = 1;
	for (i = 1; i <= n; i++) {
		cin >> x[i] >> y[i] >> r[i];
		y[i] = 0;
		ms.insert({r[i], -i});
		ms2.insert({abs(x[i] - r[i]), i});
		ms3.insert({abs(x[i] - r[i]), i});
		//y[i] = 0;
		if (y[i] != 0) {
			subtask2 = 0;
		}
	}
	if (subtask2) {
		for (i = 1; i <= n; i++) {
			if (ms.empty()) break;
			k = -(*ms.rbegin()).second;
			ms.erase(prev(ms.end()));
			ms2.erase({abs(x[k] - r[k]), k});
			ms3.erase({abs(x[k] - r[k]), k});
			ans[k] = k;
			while (ms2.size()) {
				auto it = ms2.begin();
				int j = (*it).second;
				if (r[k] + r[j] >= abs(x[j] - x[k])) {
					ans[j] = k;
					ms.erase({r[j], -j});
					ms2.erase(ms2.begin());
					ms3.erase({abs(x[j] - r[j]), j});
				}
				else break;
			}
			while (ms3.size()) {
				auto it = ms3.begin();
				int j = (*it).second;
				if (r[k] + r[j] >= abs(x[j] - x[k])) {
					ans[j] = k;
					ms.erase({r[j], -j});
					ms2.erase({abs(x[j] - r[j]), j});
					ms3.erase(ms3.begin());
				}
				else break;
			}
		}
		for (i = 1; i <= n; i++) cout << ans[i] << " ";
		return 0;
	}
	for (i = 1; i <= n; i++) {
		if (ms.empty()) break;
		k = -(*ms.rbegin()).second;
		ms.erase(prev(ms.end()));
		ans[k] = k;
		for (auto j : ms) {
			if (inter(k, -j.second)) {
				ans[-j.second] = k;
			}
			else mss.insert(j);
		}
		ms = mss;
		mss.clear();
	}
	for (i = 1; i <= n; i++) cout << ans[i] << " ";
}

• Subtask 1: 0 / 7
• Subtask 2: 0 / 12
• Subtask 3: 0 / 15
• Subtask 4: 0 / 23
• Subtask 5: 0 / 30
• Subtask 6: 0 / 13