#include <bits/stdc++.h>
#define all(x) (x).begin(), (x).end()
#define ff first
#define ss second
#define pb push_back
#define mp make_pair
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef long double ld;
typedef pair<ll, ll> pll;
typedef pair<ull, ull> pull;
typedef pair<int, int> pii;
typedef pair<ld, ld> pld;
int n;
struct node{
int x, h, v;
} a[500009];
pii seg[2000009];
multiset<pii> s[500009];
void build(int v, int tl, int tr){
if(tl == tr){
s[tl].clear();
seg[v] = {1e9, tl};
return;
}
build(2*v, tl, (tl+tr)/2);
build(2*v+1, (tl+tr)/2+1, tr);
seg[v] = min(seg[2*v], seg[2*v+1]);
}
void upd(int v, int tl, int tr, int idx){
if(idx < tl || idx > tr)
return;
if(idx == tl && idx == tr){
if(s[tl].empty())
seg[v].ff = 1e9;
else
seg[v].ff = s[tl].begin()->ff;
}
else{
upd(2*v, tl, (tl+tr)/2, idx);
upd(2*v+1, (tl+tr)/2+1, tr, idx);
seg[v] = min(seg[2*v], seg[2*v+1]);
}
}
pii get(int v, int tl, int tr, int l, int r){
if(tr < l || r < tl)
return {1e9, 1e9};
if(l <= tl && tr <= r)
return seg[v];
else{
return min(
get(2*v, tl, (tl+tr)/2, l, r),
get(2*v+1, (tl+tr)/2+1, tr, l, r)
);
}
}
int used[500009];
int ans[500009];
int mem[500009];
void solve(){
cin >> n;
int latest = 0;
for(int i = 0; i < n; ++i)
cin >> a[i].x;
for(int i = 0; i < n; ++i){
cin >> a[i].h;
if(a[i].h == 0) latest = i;
}
for(int i = 0; i < n; ++i){
cin >> a[i].v;
used[i] = -1;
}
build(1, 0, n);
multiset<pii> pr;
multiset<pii> bef;
multiset<pii, greater<pii>> kids;
for(int i = 0; i <= latest; ++i){
ans[i] = -1;
if(a[i].h)
kids.insert({a[i].v, i});
else
pr.insert({a[i].v, i});
}
int j = 0;
for(int i = latest; i < n; ++i){
if(a[i].h){
pii cur = {a[i].v, i};
pii oth = get(1, 0, n, cur.ff, n);
if(oth.ff < cur.ff){
s[oth.ss].insert(cur);
cur = *s[oth.ss].begin();
s[oth.ss].erase(s[oth.ss].begin());
upd(1, 0, n, oth.ss);
}
kids.insert(cur);
}
while(pr.size()){
pii cur = *pr.begin();
if(kids.lower_bound({cur.ff, 1e9}) != kids.end()){
pii ki = *kids.lower_bound({cur.ff, 1e9});
kids.erase(kids.lower_bound({cur.ff, 1e9}));
s[cur.ff].insert(ki);
upd(1, 0, n, cur.ff);
used[ki.ss] = cur.ss;
mem[cur.ss] = ki.ss;
pr.erase(pr.begin());
}
else break;
}
//cout << pr.size() << '\n';
while(j <= i && pr.empty()){
if(a[j].h == 0){
bef.insert({a[j].v, i});
++j;
}
else if(used[j] <= j){
if(used[j] == -1)
kids.erase({a[j].v, j});
else{
s[a[used[j]].v].erase({a[j].v, j});
upd(1, 0, n, a[used[j]].v);
}
++j;
}
else{
s[a[used[j]].v].erase({a[j].v, j});
upd(1, 0, n, a[used[j]].v);
pr.insert({a[used[j]].v, used[j]});
if(bef.lower_bound({a[j].v, 0}) != bef.end()){
int nxt = bef.lower_bound({a[j].v, 0})->ss;
bef.erase(bef.lower_bound({a[j].v, 0}));
s[a[nxt].v].erase({a[mem[nxt]].v, mem[nxt]});
upd(1, 0, n, a[nxt].v);
kids.insert({a[mem[nxt]].v, mem[nxt]});
used[mem[nxt]] = -1;
}
while(pr.size()){
pii cur = *pr.begin();
if(kids.lower_bound({cur.ff, 1e9}) != kids.end()){
pii ki = *kids.lower_bound({cur.ff, 1e9});
kids.erase(kids.lower_bound({cur.ff, 1e9}));
s[cur.ff].insert(ki);
upd(1, 0, n, cur.ff);
used[ki.ss] = cur.ss;
mem[cur.ss] = ki.ss;
pr.erase(pr.begin());
}
else break;
}
++j;
}
}
if(j == 0)
ans[i] = -1;
else{
//cout << i << ' ' << j-1 << '\n';
ans[i] = 2*a[i].x-a[j-1].x;
}
}
for(int i = n-1; i >= 1; --i)
if(a[i].x == a[i-1].x)
ans[i-1] = ans[i];
for(int i = 0; i < n; ++i)
cout << ans[i] << ' ';
cout << '\n';
}
int main() {
ios_base::sync_with_stdio(false);
cin.tie(nullptr);
//freopen("in.txt", "r", stdin);
//freopen("out.txt", "w", stdout);
int t;
cin >> t;
while(t--){
solve();
}
}
| # |
결과 |
실행 시간 |
메모리 |
Grader output |
| 1 |
Incorrect |
17 ms |
23808 KB |
Output isn't correct |
| 2 |
Incorrect |
17 ms |
23920 KB |
Output isn't correct |
| 3 |
Incorrect |
28 ms |
24064 KB |
Output isn't correct |
| 4 |
Incorrect |
44 ms |
24192 KB |
Output isn't correct |
| 5 |
Incorrect |
84 ms |
24696 KB |
Output isn't correct |
| 6 |
Incorrect |
174 ms |
25336 KB |
Output isn't correct |
| 7 |
Incorrect |
334 ms |
26860 KB |
Output isn't correct |
| 8 |
Incorrect |
405 ms |
28420 KB |
Output isn't correct |
| 9 |
Incorrect |
504 ms |
32632 KB |
Output isn't correct |
| 10 |
Incorrect |
800 ms |
33372 KB |
Output isn't correct |
