#include <bits/stdc++.h>
using namespace std;
struct Container {
int index;
unordered_set<int> colors;
Container* parent;
vector<Container*> children;
Container(Container* p, int i): parent(p), index(i) {}
};
struct Event {
int index;
int color;
int x_pos;
int y_low, y_high;
bool close;
Event(int col, int x, int yl, int yh, bool c, int i):
color(col), x_pos(x), y_low(yl), y_high(yh), close(c), index(i) {}
};
struct Node {
Node *l = 0, *r = 0;
int lo, hi;
Container* mset = NULL;
Container* val = NULL;
Node(int lo,int hi): lo(lo), hi(hi) {}
Container* query(int L, int R) {
if (R <= lo || hi <= L) return NULL;
if (L <= lo && hi <= R) return val;
push();
Container* result;
if((result = l->query(L, R)) != NULL)
return result;
else if((result = r->query(L, R)) != NULL)
return result;
else
return NULL;
}
void set(int L, int R, Container* x) {
if (R <= lo || hi <= L) return;
if (L <= lo && hi <= R) mset = val = x;
else {
push(), l->set(L, R, x), r->set(L, R, x);
}
}
void push() {
if (!l) {
int mid = lo + (hi - lo)/2;
l = new Node(lo, mid); r = new Node(mid, hi);
}
if (mset != NULL)
l->set(lo,hi,mset), r->set(lo,hi,mset), mset = NULL;
}
};
// void resolve(Container* node, vector<int> &colorCounts){
// for(Container* child : node->children)
// resolve(child, colorCounts);
// if(node->index != -1)
// colorCounts[node->index] = node->colors.size();
// }
int main(){
cin.tie(NULL)->sync_with_stdio(false);
int n, m;
cin >> n >> m;
vector<Event> actions;
for(int i = 0; i < n; i++){
int lx, ly, ux, uy;
cin >> lx >> ly >> ux >> uy;
actions.emplace_back(-1, lx, ly, uy, false, i);
actions.emplace_back(-1, ux, ly, uy, true, i);
}
for(int i = 0; i < m; i++){
int x, y, c;
cin >> x >> y >> c;
actions.emplace_back(c, x, y, y, true, i);
}
auto actionComp = [](const auto &a, const auto &b){
if(a.x_pos == b.x_pos){
if(a.color == -1 && b.color == -1)
return a.y_low < b.y_low;
else if(a.color == -1 && b.color != -1)
return !b.close;
else if(a.color != -1 && b.color == -1)
return b.close;
else
return a.y_low < b.y_low;
}else
return a.x_pos < b.x_pos;
};
sort(actions.begin(), actions.end(), actionComp);
Container root(NULL, -1);
Node* containerQuery = new Node(-10, 1e9+10);
vector<int> colorCounts(n, 0);
containerQuery->set(0, (1e9 + 1) + 1, &root);
for(const Event event : actions){
if(event.color == -1){
if(event.close){
Container* child = containerQuery->query(event.y_low, event.y_high + 1);
cout << "Closing: " << child->index << "\n";
colorCounts[child->index] = child->colors.size();
if(child->parent->colors.size() < child->colors.size())
swap(child->parent->colors, child->colors);
child->parent->colors.insert(child->colors.begin(), child->colors.end());
containerQuery->set(event.y_low, event.y_high + 1, child->parent);
}else{
Container* parent = containerQuery->query(event.y_low, event.y_high + 1);
parent->children.push_back(new Container(parent, event.index));
cout << "Opening: " << event.index << "\n";
containerQuery->set(event.y_low, event.y_high + 1, parent->children.back());
}
}else{
Container* sheet = containerQuery->query(event.y_low, event.y_high + 1);
cout << "Inserting into: " << sheet->index << "\n";
sheet->colors.insert(event.color);
}
}
for(const int &count : colorCounts)
cout << count << "\n";
}
Compilation message
plahte.cpp: In constructor 'Container::Container(Container*, int)':
plahte.cpp:8:16: warning: 'Container::parent' will be initialized after [-Wreorder]
8 | Container* parent;
| ^~~~~~
plahte.cpp:6:9: warning: 'int Container::index' [-Wreorder]
6 | int index;
| ^~~~~
plahte.cpp:11:5: warning: when initialized here [-Wreorder]
11 | Container(Container* p, int i): parent(p), index(i) {}
| ^~~~~~~~~
plahte.cpp: In constructor 'Event::Event(int, int, int, int, bool, int)':
plahte.cpp:19:10: warning: 'Event::close' will be initialized after [-Wreorder]
19 | bool close;
| ^~~~~
plahte.cpp:15:9: warning: 'int Event::index' [-Wreorder]
15 | int index;
| ^~~~~
plahte.cpp:21:5: warning: when initialized here [-Wreorder]
21 | Event(int col, int x, int yl, int yh, bool c, int i):
| ^~~~~
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
133 ms |
69048 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
149 ms |
72552 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
249 ms |
120136 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
418 ms |
185024 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
410 ms |
183168 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |