#include<iostream>
#include<set>
#include<vector>
#include<unordered_set>
using namespace std;
typedef long long ll;
const int TREE_SIZE=1<<30;
vector<int>elements;
struct Node;
Node* insert_fn(Node* old_node,int x,int q);
struct Node{
int val=0;
Node*ch1=nullptr,*ch2=nullptr;
void collect_fn(int l,int r){
if(l==r-1){
if(val!=0)
elements.push_back(l);
return;
}
int m=(l+r)/2;
if(ch1)
ch1->collect_fn(l,m);
if(ch2)
ch2->collect_fn(m,r);
}
vector<int>collect(){
elements.clear();
collect_fn(0,TREE_SIZE);
return elements;
}
~Node(){
delete ch1;
delete ch2;
}
};
Node* insert_fn(Node* old_node,int x,int q,int l,int r){
if(x<l||r<=x)
return old_node;
Node* node=new Node;
if(old_node){
node->ch1=old_node->ch1;
node->ch2=old_node->ch2;
node->val=old_node->val;
}
if(l==r-1){
node->val+=q;
return node;
}
int m=(l+r)/2;
node->ch1=insert_fn(node->ch1,x,q,l,m);
node->ch2=insert_fn(node->ch2,x,q,m,r);
return node;
}
Node* insert(Node* old_node,int x,int q){
return insert_fn(old_node,x,q,0,TREE_SIZE);
}
// time travelling multiset
class TTMS{
vector<int>days;
vector<Node*>sets;
Node* set;
public:
TTMS(){
//days={-1};
//sets={new Node};
set=new Node;
}
void add_new_day(int day){
//days.push_back(day);
//sets.push_back(sets.back());
}
void insert(int el){
//sets.back()=::insert(sets.back(),el,1);
set=::insert(set,el,1);
}
void remove(int el){
//sets.back()=::insert(sets.back(),el,-1);
set=::insert(set,el,-1);
}
vector<int> get(int day){
/*int i=0;
while(i<(int)days.size()-1&&days[i+1]<=day)
i++;
return sets[i]->collect();*/
return set->collect();
}
};
int get_closest(vector<int>a,vector<int>b){
int res=1e9;
for(int i:a)
for(int j:b){
res=min(res,abs(i-j));
}
return res;
}
int n;
int heights[100000];
TTMS ttms[100000];
void init(int N, int D, int H[]) {
n=N;
for(int i=0;i<n;i++)
heights[i]=H[i];
}
ll to(int a,int b){
if(a<b)
swap(a,b);
return a+((ll)b<<32);
}
void curseChanges(int U, int A[], int B[]) {
unordered_set<ll>conns;
for(int day=0;day<U;day++){
int a=A[day],b=B[day];
ll conn=to(a,b);
ttms[a].add_new_day(day+1);
ttms[b].add_new_day(day+1);
if(conns.find(conn)==conns.end()){
conns.insert(conn);
ttms[a].insert(heights[b]);
ttms[b].insert(heights[a]);
}else{
conns.erase(conn);
ttms[a].remove(heights[b]);
ttms[b].remove(heights[a]);
}
}
}
int question(int x, int y, int v) {
vector<int>a=ttms[x].get(v);
vector<int>b=ttms[y].get(v);
return get_closest(a,b);
}
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
5 ms |
8912 KB |
Incorrect |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
8 ms |
10880 KB |
Incorrect |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Runtime error |
461 ms |
262144 KB |
Execution killed with signal 9 |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Runtime error |
425 ms |
262144 KB |
Execution killed with signal 9 |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
38 ms |
28792 KB |
Incorrect |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
5 ms |
8912 KB |
Incorrect |
2 |
Halted |
0 ms |
0 KB |
- |