Submission #1306878

#TimeUsernameProblemLanguageResultExecution timeMemory
1306878chrisvilchesPlahte (COCI17_plahte)C++20
0 / 160
55 ms5764 KiB
#include <bits/stdc++.h>
using namespace std;
using pii = pair<int, int>;

struct BIT {
  BIT(const int size) : n(size + 1), A(n, 0) {}

  void range_update(const int i, const int j, const int v) {
    update(i, v);
    update(j + 1, -v);
  }

  int query(int i) const {
    i++;
    int sum = 0;
    for (; i > 0; i -= i & -i) sum += A[i];
    return sum;
  }

 private:
  const int n;
  vector<int> A;
  void update(int i, const int v) {
    i++;
    for (; i < n; i += i & -i) A[i] += v;
  }
};

int main() {
  ios_base::sync_with_stdio(false);
  cin.tie(NULL);

  int n, m;
  while (cin >> n >> m) {
    vector<pair<int, int>> events;
    vector<tuple<int, int, int, int>> rectangles;
    vector<pair<int, int>> points(m);

    for (int i = 0; i < n; i++) {
      int x1, y1, x2, y2;
      cin >> x1 >> y1 >> x2 >> y2;
      rectangles.emplace_back(x1, y1, x2, y2);
      events.emplace_back(x1, i);
      events.emplace_back(x2, ~i);
    }

    for (pair<int, int>& p : points) {
      int tag;
      cin >> p.first >> p.second;
      cin >> tag;
    }

    sort(points.begin(), points.end());
    sort(events.begin(), events.end());

    const int bit_size = 100;
    BIT bit(bit_size);

    bit.range_update(0, bit_size - 1, -1);

    vector<int> delta(n);

    // size_t k = 0;

    vector<vector<int>> graph(n);

    for (const auto& [x, i] : events) {
      const auto [_, y1, __, y2] = rectangles[i < 0 ? ~i : i];

      if (i < 0) {
        bit.range_update(y1, y2, -delta[~i]);
        continue;
      }

      const int curr_value = bit.query(y1);
      delta[i] = i - curr_value;

      bit.range_update(y1, y2, delta[i]);

      if (curr_value == -1) {
        // Doesn't point to any other node.
      } else {
        graph[i].emplace_back(curr_value);
      }
    }

    // TODO: DFS and collect results. Warning: It's a DAG, will it get TLE?? Do a DP from
    // the leaves?

    vector<int> in_deg(n, 0);

    for (const vector<int>& adj : graph) {
      for (const int v : adj) {
        in_deg[v]++;
      }
    }

    set<pair<int, int>> bullets{points.begin(), points.end()};

    vector<int> bullets_in_rect(n, 0);

    const function<void(int)> dfs = [&](const int u) {
      vector<pii> inside;
      const auto [x1, y1, x2, y2] = rectangles[u];
      for (const pii& b : bullets) {
        if (x1 <= b.first && b.first <= x2 && y1 <= b.second && b.second <= y2) {
          inside.emplace_back(b);
        }
      }

      bullets_in_rect[u] += inside.size();

      for (const pii& b : inside) {
        assert(bullets.count(b));
        bullets.erase(b);
      }

      for (const int v : graph[u]) {
        bullets_in_rect[v] += bullets_in_rect[u];
        dfs(v);
      }
    };

    for (int i = 0; i < n; i++) {
      if (in_deg[i] != 0) continue;
      dfs(i);
    }

    for (const int x : bullets_in_rect) {
      cout << x << endl;
    }
  }
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...