// And you curse yourself for things you never done
#include<bits/stdc++.h>
#define F first
#define S second
#define PB push_back
#define sz(s) int((s).size())
#define bit(n,k) (((n)>>(k))&1)
using namespace std;
typedef long long ll;
typedef pair<int,int> pii;
const int maxn = 3e5 + 10, mod = 1e9 + 7, inf = 1e9 + 10;
const ll INF = 1e18;
pair<pii, int> p[maxn];
int arr[maxn], ans[maxn];
pii operator - (pii a, pii b){
return {a.F - b.F, a.S - b.S};
}
pii operator + (pii a, pii b){
return {a.F + b.F, a.S + b.S};
}
ll operator ^ (pii a, pii b){
return 1ll * a.F * b.F + 1ll * a.S * b.S;
}
bool ok(int i, int j){
return 1ll * (p[j].S + p[i].S) * (p[j].S + p[i].S) >= ((p[i].F - p[j].F) ^ (p[i].F - p[j].F));
}
vector<int> vx, vy;
vector<int> val, fs;
vector<int> stock;
void push(int &lst, int x){
fs.PB(lst);
lst = sz(val);
val.PB(x);
}
struct node{
map<int, int> mp;
void add(int pos, int x){
mp[pos] = x;
}
void ask(int i){
int L = p[i].F.S - p[i].S, R = p[i].F.S + p[i].S;
auto it = mp.lower_bound(R);
if(it != mp.end())
stock.PB(it->S);
while(it != mp.begin() && (it->F) > L)
--it, stock.PB(it->S);
if(it != mp.begin())
--it, stock.PB(it->S);
}
};
struct node2{
node2 *L = 0, *R = 0;
node* seg = 0;
void add(int f, int s, int pos, int x, int l = 0, int r = sz(vx)-1){
if(f <= vx[l] && vx[r] <= s){
if(!seg)
seg = new node();
seg->add(pos, x);
return;
}
if(r-l == 1)
return;
int mid = (l+r) >> 1;
if(max(f, vx[l]) < min(s, vx[mid])){
if(!L)
L = new node2();
L->add(f, s, pos, x, l, mid);
}
if(max(f, vx[mid]) < min(s, vx[r])){
if(!R)
R = new node2();
R->add(f, s, pos, x, mid, r);
}
}
void ask(int pos, int pos2, int l = 0, int r = sz(vx)-1){
if(seg)
seg->ask(pos2);
if(r-l == 1)
return;
int mid = (l+r) >> 1;
if(pos < vx[mid] && L)
L->ask(pos, pos2, l, mid);
if(vx[mid] <= pos && R)
R->ask(pos, pos2, mid, r);
}
};
int main(){
ios_base::sync_with_stdio(false); cin.tie(0); cout.tie();
node2* root = new node2();
int n;
cin >> n;
for(int i = 0; i < n; i++){
cin >> p[i].F.F >> p[i].F.S >> p[i].S;
vx.PB(p[i].F.F - p[i].S);
vx.PB(p[i].F.F + p[i].S + 1);
vy.PB(p[i].F.S - p[i].S);
vy.PB(p[i].F.S + p[i].S + 1);
arr[i] = i;
}
auto cmp = [](int i, int j){
return (p[i].S == p[j].S ? i > j : p[i].S < p[j].S);
};
sort(arr, arr + n, cmp);
sort(vx.begin(), vx.end());
vx.resize( unique(vx.begin(), vx.end()) - vx.begin() );
sort(vy.begin(), vy.end());
vy.resize( unique(vy.begin(), vy.end()) - vy.begin() );
for(int i = n-1; i >= 0; i--){
int I = arr[i];
stock.clear();
for(int w = -1; w <= 1; w++){
root->ask(p[I].F.F + w * p[I].S, I); // kafie?
}
for(int id : stock){
if(ok(I, id)){
if(ans[I] == 0 || cmp(ans[I] - 1, id))
ans[I] = id + 1;
}
}
if(ans[I] == 0){
ans[I] = I + 1;
root->add(p[I].F.F - p[I].S, p[I].F.F + p[I].S + 1, p[I].F.S, I);
}
}
for(int i = 0; i < n; i++){
cout << ans[i] << " ";
}
return cout << endl, 0;
}
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
0 ms |
384 KB |
Output is correct |
2 |
Correct |
0 ms |
384 KB |
Output is correct |
3 |
Correct |
0 ms |
384 KB |
Output is correct |
4 |
Correct |
0 ms |
384 KB |
Output is correct |
5 |
Correct |
0 ms |
384 KB |
Output is correct |
6 |
Correct |
0 ms |
384 KB |
Output is correct |
7 |
Correct |
0 ms |
384 KB |
Output is correct |
8 |
Incorrect |
1 ms |
384 KB |
Output isn't correct |
9 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
338 ms |
13256 KB |
Output is correct |
2 |
Correct |
337 ms |
13588 KB |
Output is correct |
3 |
Correct |
347 ms |
12992 KB |
Output is correct |
4 |
Correct |
322 ms |
13300 KB |
Output is correct |
5 |
Correct |
461 ms |
17624 KB |
Output is correct |
6 |
Correct |
648 ms |
52556 KB |
Output is correct |
7 |
Correct |
534 ms |
23376 KB |
Output is correct |
8 |
Correct |
582 ms |
32192 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
1 ms |
384 KB |
Output is correct |
2 |
Correct |
457 ms |
53216 KB |
Output is correct |
3 |
Correct |
1518 ms |
156568 KB |
Output is correct |
4 |
Correct |
1500 ms |
156040 KB |
Output is correct |
5 |
Correct |
1917 ms |
141104 KB |
Output is correct |
6 |
Correct |
747 ms |
58068 KB |
Output is correct |
7 |
Correct |
280 ms |
28000 KB |
Output is correct |
8 |
Correct |
31 ms |
4468 KB |
Output is correct |
9 |
Correct |
1646 ms |
157640 KB |
Output is correct |
10 |
Correct |
2547 ms |
124824 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
1243 ms |
145528 KB |
Output is correct |
2 |
Correct |
677 ms |
74244 KB |
Output is correct |
3 |
Incorrect |
1789 ms |
30960 KB |
Output isn't correct |
4 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
0 ms |
384 KB |
Output is correct |
2 |
Correct |
0 ms |
384 KB |
Output is correct |
3 |
Correct |
0 ms |
384 KB |
Output is correct |
4 |
Correct |
0 ms |
384 KB |
Output is correct |
5 |
Correct |
0 ms |
384 KB |
Output is correct |
6 |
Correct |
0 ms |
384 KB |
Output is correct |
7 |
Correct |
0 ms |
384 KB |
Output is correct |
8 |
Incorrect |
1 ms |
384 KB |
Output isn't correct |
9 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
0 ms |
384 KB |
Output is correct |
2 |
Correct |
0 ms |
384 KB |
Output is correct |
3 |
Correct |
0 ms |
384 KB |
Output is correct |
4 |
Correct |
0 ms |
384 KB |
Output is correct |
5 |
Correct |
0 ms |
384 KB |
Output is correct |
6 |
Correct |
0 ms |
384 KB |
Output is correct |
7 |
Correct |
0 ms |
384 KB |
Output is correct |
8 |
Incorrect |
1 ms |
384 KB |
Output isn't correct |
9 |
Halted |
0 ms |
0 KB |
- |