bool home = 0;
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N = (int) 3e5 + 7;
int n;
struct Circle {
int x;
int y;
int r;
};
bool doIntersect(Circle a, Circle b) {
/// dist <= a.r + b.r
/// dist ^ 2 <= (a.r + b.r) ^ 2
return 1LL * (a.x - b.x) * (a.x - b.x) + 1LL * (a.y - b.y) * (a.y - b.y) <= 1LL * (a.r + b.r) * (a.r + b.r);
}
Circle circles[N];
int ret[N];
map<pair<int, int>, set<int>> guysAtPosition;
bool cmp(int i, int j) {
if (circles[i].r == circles[j].r) {
return i < j;
} else {
return circles[i].r > circles[j].r;
}
}
set<int, decltype(cmp)*> s(cmp);
int r_divide_glob;
void place(int r_divide, set<int, decltype(cmp)*>& s) {
r_divide_glob = r_divide;
guysAtPosition.clear();
for (auto &i : s) {
guysAtPosition[{circles[i].x / r_divide, circles[i].y / r_divide}].insert(i);
}
}
void elim(int i, set<int, decltype(cmp)*>& s) {
int x = circles[i].x / r_divide_glob, y = circles[i].y / r_divide_glob;
for (int x2 = x - 2; x2 <= x + 2; x2++) {
for (int y2 = y - 2; y2 <= y + 2; y2++) {
pair<int, int> it = {x2, y2};
vector<int> dl;
for (auto &j : guysAtPosition[it]) {
if (doIntersect(circles[i], circles[j])) {
ret[j] = i;
s.erase(j);
dl.push_back(j);
}
}
for (auto &j : dl) {
guysAtPosition[it].erase(j);
}
}
}
}
int main() {
if (home == 0) {
ios::sync_with_stdio(0); cin.tie(0); cout.tie(0);
} else {
freopen ("input.txt", "r", stdin);
}
cin >> n;
for (int i = 1; i <= n; i++) {
cin >> circles[i].x >> circles[i].y >> circles[i].r;
}
for (int i = 1; i <= n; i++) {
s.insert(i);
}
place(circles[1].r, s);
while (!s.empty()) {
int i = *s.begin();
elim(i, s);
}
for (int i = 1; i <= n; i++) {
cout << ret[i] << " ";
}
cout << "\n";
return 0;
}
Compilation message
circle_selection.cpp: In function 'int main()':
circle_selection.cpp:76:13: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
76 | freopen ("input.txt", "r", stdin);
| ~~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
0 ms |
212 KB |
Output is correct |
2 |
Correct |
0 ms |
212 KB |
Output is correct |
3 |
Correct |
1 ms |
340 KB |
Output is correct |
4 |
Incorrect |
1 ms |
340 KB |
Output isn't correct |
5 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
812 ms |
36172 KB |
Output is correct |
2 |
Correct |
880 ms |
42012 KB |
Output is correct |
3 |
Incorrect |
885 ms |
41880 KB |
Output isn't correct |
4 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
1 ms |
340 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Execution timed out |
3096 ms |
679400 KB |
Time limit exceeded |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
0 ms |
212 KB |
Output is correct |
2 |
Correct |
0 ms |
212 KB |
Output is correct |
3 |
Correct |
1 ms |
340 KB |
Output is correct |
4 |
Incorrect |
1 ms |
340 KB |
Output isn't correct |
5 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
0 ms |
212 KB |
Output is correct |
2 |
Correct |
0 ms |
212 KB |
Output is correct |
3 |
Correct |
1 ms |
340 KB |
Output is correct |
4 |
Incorrect |
1 ms |
340 KB |
Output isn't correct |
5 |
Halted |
0 ms |
0 KB |
- |