#include <iostream> #include <vector> #define dbg //printf using namespace std; struct Node { int from, to; int color; Node *left, *right, *parent; int middle() const { return (from + to)/2; } Node(int from, int to) { dbg("(%d %d)\n", from, to); this->from = from; this->to = to; color = 0; if (from != to) { left = new Node(from, middle()); left->parent = this; right = new Node(middle()+1, to); right->parent = this; } else { left = nullptr; right = nullptr; } } void mark(int a, int b, int value) { dbg("(%d %d) ++ %d\n", a, b, value); if (a == from && b == to) { dbg("marking!\n"); color |= value; } else if (b <= middle()) { left->mark(a, b, value); } else if (a > middle()) { right->mark(a, b, value); } else { left->mark(a, middle(), value); right->mark(middle()+1, b, value); } } int what_col(int where) { // dbg("where: (%d %d) %d ", from, to, where, color); if (where < from || where > to) { // dbg("-> %d\n", 0); return 0; } if (from == to && to == where) { // dbg("-> %d\n", this->color); return this->color; } // dbg("-> %d\n", this->color); return this->color | this->left->what_col(where) | this->right->what_col(where); } }; int main() { int n, m; cin >> n >> m; // make n a power of 2 dbg("%d\n", n); Node root(1, n); for (int i = 0; i < m; ++i) { int from, to, what; scanf("%d %d %d", &from, &to, &what); root.mark(from, to, (1 << (what-1))); dbg("\n\n"); } int counter = 0; for (int i = 1; i <= n; ++i) { auto c = root.what_col(i); dbg("%d: %d\n", i, c); if (c == 3) { counter++; } } cout << counter << endl; return 0; }
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 | #include <iostream> #include <vector> #define dbg //printf using namespace std; struct Node { int from, to; int color; Node *left, *right, *parent; int middle() const { return (from + to)/2; } Node(int from, int to) { dbg("(%d %d)\n", from, to); this->from = from; this->to = to; color = 0; if (from != to) { left = new Node(from, middle()); left->parent = this; right = new Node(middle()+1, to); right->parent = this; } else { left = nullptr; right = nullptr; } } void mark(int a, int b, int value) { dbg("(%d %d) ++ %d\n", a, b, value); if (a == from && b == to) { dbg("marking!\n"); color |= value; } else if (b <= middle()) { left->mark(a, b, value); } else if (a > middle()) { right->mark(a, b, value); } else { left->mark(a, middle(), value); right->mark(middle()+1, b, value); } } int what_col(int where) { // dbg("where: (%d %d) %d ", from, to, where, color); if (where < from || where > to) { // dbg("-> %d\n", 0); return 0; } if (from == to && to == where) { // dbg("-> %d\n", this->color); return this->color; } // dbg("-> %d\n", this->color); return this->color | this->left->what_col(where) | this->right->what_col(where); } }; int main() { int n, m; cin >> n >> m; // make n a power of 2 dbg("%d\n", n); Node root(1, n); for (int i = 0; i < m; ++i) { int from, to, what; scanf("%d %d %d", &from, &to, &what); root.mark(from, to, (1 << (what-1))); dbg("\n\n"); } int counter = 0; for (int i = 1; i <= n; ++i) { auto c = root.what_col(i); dbg("%d: %d\n", i, c); if (c == 3) { counter++; } } cout << counter << endl; return 0; } |