#include <bits/stdc++.h>
using namespace std;
using ll=long long;
#define int ll
using ari2=array<int,2>;
using ari3=array<int,3>;
#define vt vector
#define size(x) (int((x).size()))
#define all(x) begin(x), end(x)
#define REP(a,b,c,d) for (int a=(b);(d)>0?a<=(c):a>=(c);a+=(d))
#define FOR(a,b,c) REP(a,b,c,1)
#define ROF(a,b,c) REP(a,b,c,-1)
signed main() {
#ifndef DEBUG
ios_base::sync_with_stdio(false);
cin.tie(nullptr);
#endif
int N;
cin >> N;
vt<int> X(N), Y(N), R(N);
FOR(i, 0, N-1)
cin >> X[i] >> Y[i] >> R[i];
vt<int> ord(N);
iota(all(ord), 0);
sort(all(ord), [&](const int &a, const int &b) {
return ari2{R[a], b} > ari2{R[b], a};
});
auto good = [&](int i, int j) {
auto sq = [&](int x) {
return 1ll * x * x;
};
return sq(X[i]-X[j]) + sq(Y[i]-Y[j]) <= sq(R[i]+R[j]);
};
vt<int> ans(N, -1);
for (int r=1<<30; r>0; r>>=1) {
vt<ari3> points;
FOR(i, 0, N-1)
if (ans[i] < 0)
points.push_back({X[i]/r, Y[i]/r, i});
sort(all(points));
for (int i : ord) {
if (ans[i]>=0 || R[i] < r/2 || R[i] > r)
continue;
ans[i] = i;
int x = X[i] / r;
int y = Y[i] / r;
FOR(d, -2, 2)
FOR(j, lower_bound(all(points), ari3{x+d, y-2, 0}) - begin(points), size(points)-1) {
auto [xj, yj, ij] = points[j];
if (xj != x+d || abs(yj-y) > 2)
break;
if (good(i, ij))
ans[ij] = i;
}
}
}
FOR(i, 0, N-1)
cout << (ans[i]<0 ? i : ans[i]) + 1 << "\n "[i+1<N], assert(ans[i] >= 0);
}
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
0 ms |
348 KB |
Output is correct |
2 |
Correct |
0 ms |
348 KB |
Output is correct |
3 |
Correct |
0 ms |
348 KB |
Output is correct |
4 |
Correct |
0 ms |
348 KB |
Output is correct |
5 |
Correct |
1 ms |
348 KB |
Output is correct |
6 |
Correct |
0 ms |
348 KB |
Output is correct |
7 |
Correct |
1 ms |
348 KB |
Output is correct |
8 |
Correct |
0 ms |
348 KB |
Output is correct |
9 |
Correct |
1 ms |
348 KB |
Output is correct |
10 |
Incorrect |
0 ms |
348 KB |
Output isn't correct |
11 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
188 ms |
32920 KB |
Output is correct |
2 |
Incorrect |
244 ms |
32576 KB |
Output isn't correct |
3 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
1 ms |
348 KB |
Output is correct |
2 |
Correct |
291 ms |
9808 KB |
Output is correct |
3 |
Correct |
1049 ms |
35028 KB |
Output is correct |
4 |
Correct |
1066 ms |
33544 KB |
Output is correct |
5 |
Correct |
955 ms |
31984 KB |
Output is correct |
6 |
Correct |
451 ms |
16632 KB |
Output is correct |
7 |
Correct |
228 ms |
8696 KB |
Output is correct |
8 |
Correct |
44 ms |
2324 KB |
Output is correct |
9 |
Correct |
1078 ms |
32980 KB |
Output is correct |
10 |
Correct |
850 ms |
31980 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
1087 ms |
32576 KB |
Output is correct |
2 |
Correct |
1491 ms |
32536 KB |
Output is correct |
3 |
Incorrect |
591 ms |
32548 KB |
Output isn't correct |
4 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
0 ms |
348 KB |
Output is correct |
2 |
Correct |
0 ms |
348 KB |
Output is correct |
3 |
Correct |
0 ms |
348 KB |
Output is correct |
4 |
Correct |
0 ms |
348 KB |
Output is correct |
5 |
Correct |
1 ms |
348 KB |
Output is correct |
6 |
Correct |
0 ms |
348 KB |
Output is correct |
7 |
Correct |
1 ms |
348 KB |
Output is correct |
8 |
Correct |
0 ms |
348 KB |
Output is correct |
9 |
Correct |
1 ms |
348 KB |
Output is correct |
10 |
Incorrect |
0 ms |
348 KB |
Output isn't correct |
11 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
0 ms |
348 KB |
Output is correct |
2 |
Correct |
0 ms |
348 KB |
Output is correct |
3 |
Correct |
0 ms |
348 KB |
Output is correct |
4 |
Correct |
0 ms |
348 KB |
Output is correct |
5 |
Correct |
1 ms |
348 KB |
Output is correct |
6 |
Correct |
0 ms |
348 KB |
Output is correct |
7 |
Correct |
1 ms |
348 KB |
Output is correct |
8 |
Correct |
0 ms |
348 KB |
Output is correct |
9 |
Correct |
1 ms |
348 KB |
Output is correct |
10 |
Incorrect |
0 ms |
348 KB |
Output isn't correct |
11 |
Halted |
0 ms |
0 KB |
- |