#include "bits/stdc++.h"
using namespace std;
#define ar array
const double eps = 1e-9;
struct poi{
int x, y, r;
};
signed main(){
ios::sync_with_stdio(0); cin.tie(0);
int n; cin>>n;
vector<poi> a(n);
vector<set<ar<int, 2>>> pp;
vector<int> used(n), tot;
int R;
auto get = [&](int x){ return (x / R) - (x < 0); };
auto rec = [&](){
pp.clear(); tot.clear();
for(int i=0;i<n;i++) {
if(used[i]) continue;
tot.push_back(get(a[i].x));
}
sort(tot.begin(), tot.end());
tot.erase(unique(tot.begin(), tot.end()), tot.end());
pp.resize(tot.size());
for(int i=0;i<n;i++){
if(used[i]) continue;
auto j = lower_bound(tot.begin(), tot.end(), get(a[i].x)) - tot.begin();
assert(j < (int)tot.size());
pp[j].insert({a[i].y, i});
}
};
vector<int> p(n);
for(int i=0;i<n;i++){
p[i] = i;
cin>>a[i].x>>a[i].y>>a[i].r;
}
sort(p.begin(), p.end(), [&](int i, int j){
return (a[i].r > a[j].r);
});
auto sq = [&](int x) { return x * 1ll * x; };
auto dis = [&](int i, int j) -> double{
return sqrt(sq(a[i].x - a[j].x) + sq(a[i].y - a[j].y));
};
auto check = [&](int i, int j){
return (a[i].r + a[j].r > dis(i, j));
};
//~ cout<<"here"<<endl;
R = a[p[0]].r;
rec();
//~ cout<<"here"<<endl;
for(auto i : p){
if(used[i]) continue;
if(a[i].r * 2 <= R) R = a[i].r, rec();
int l = lower_bound(tot.begin(), tot.end(), get(a[i].x) - 2) - tot.begin();
int r = upper_bound(tot.begin(), tot.end(), get(a[i].x) + 2) - tot.begin();
int lx = (get(a[i].y) - 2) * R;
int rx = (get(a[i].y) + 3) * R;
for(int j=l;j<r;j++){
auto it = pp[j].lower_bound({lx, -1});
vector<ar<int, 2>> er;
while(it != pp[j].end() && (*it)[0] < rx){
if(check((*it)[1], i)){
er.push_back(*it);
} it++;
}
for(auto x : er) pp[j].erase(x), used[x[1]] = i + 1;
}
}
for(int i=0;i<n;i++) cout<<used[i]<<" ";
cout<<"\n";
}
/*
11
9 9 2
13 2 1
11 8 2
3 3 2
3 12 1
12 14 1
9 8 5
2 8 2
5 2 1
14 4 2
14 14 1
*/
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1 ms |
212 KB |
Output is correct |
2 |
Correct |
1 ms |
212 KB |
Output is correct |
3 |
Correct |
1 ms |
212 KB |
Output is correct |
4 |
Correct |
1 ms |
316 KB |
Output is correct |
5 |
Correct |
1 ms |
284 KB |
Output is correct |
6 |
Incorrect |
1 ms |
324 KB |
Output isn't correct |
7 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
276 ms |
26016 KB |
Output is correct |
2 |
Correct |
361 ms |
25684 KB |
Output is correct |
3 |
Correct |
299 ms |
26164 KB |
Output is correct |
4 |
Incorrect |
427 ms |
26416 KB |
Output isn't correct |
5 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1 ms |
212 KB |
Output is correct |
2 |
Correct |
207 ms |
7912 KB |
Output is correct |
3 |
Correct |
811 ms |
22896 KB |
Output is correct |
4 |
Correct |
838 ms |
24908 KB |
Output is correct |
5 |
Incorrect |
909 ms |
21568 KB |
Output isn't correct |
6 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
557 ms |
27676 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1 ms |
212 KB |
Output is correct |
2 |
Correct |
1 ms |
212 KB |
Output is correct |
3 |
Correct |
1 ms |
212 KB |
Output is correct |
4 |
Correct |
1 ms |
316 KB |
Output is correct |
5 |
Correct |
1 ms |
284 KB |
Output is correct |
6 |
Incorrect |
1 ms |
324 KB |
Output isn't correct |
7 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1 ms |
212 KB |
Output is correct |
2 |
Correct |
1 ms |
212 KB |
Output is correct |
3 |
Correct |
1 ms |
212 KB |
Output is correct |
4 |
Correct |
1 ms |
316 KB |
Output is correct |
5 |
Correct |
1 ms |
284 KB |
Output is correct |
6 |
Incorrect |
1 ms |
324 KB |
Output isn't correct |
7 |
Halted |
0 ms |
0 KB |
- |