BOJ7626: 직사각형

https://www.acmicpc.net/problem/7626

  • 어제에 이어서 직사각형의 총 넓이 문제
  • 이번에는 좌표 공간이 넓고, 좌표 압축이 필요하다.
  • 좌표 압축을 Unordered Map으로 했더니 매우 느리다. 좌표 압축 기법은 나중에 따로 공부해보자.
  • 이전에 숫자 한칸 한칸을 넓이라고 생각할 때와는 다르게, 각 좌표를 점으로 생각하고 Segment 트리를 쪼개야 한다.
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
#include <bits/stdc++.h>
using namespace std;

struct Tree {
  int val, cnt;
};
unordered_map<int, int> tr, rtr;

class SegmentTree {
  vector<Tree> tree;
  void refresh(int node, int start, int end) {
    if(tree[node].cnt)
      tree[node].val = rtr[end] - rtr[start];
    else if(start + 1 != end)
      tree[node].val = tree[2*node].val + tree[2*node+1].val;
    else
      tree[node].val = 0;
  }

public:
  SegmentTree(int n): tree(4*n) {}
  void update(int node, int start, int end, int l, int r, int v) {
    if(r <= start || end <= l) return;
    if(l <= start && end <= r) {
      tree[node].cnt += v;
      refresh(node, start, end);
      return;
    }
    int m = (start + end) / 2;
    update(2*node, start, m, l, r, v);
    update(2*node+1, m, end, l, r, v);
    refresh(node, start, end);
  }
  int query() {
    return tree[1].val;
  }
};

enum Cmd { ADD = 1, DEL = -1 };
struct Event {
  Cmd cmd;
  int x, y1, y2;
  bool operator<(const Event b) const {
    return x < b.x;
  }
};

int main() {
  int N;
  vector<Event> events;
  set<int> yCoord; 
  scanf("%d", &N);
  events.reserve(2*N);
  for (int i = 0; i < N; i++) {
    int x1, y1, x2, y2;
    scanf("%d%d%d%d", &x1, &x2, &y1, &y2);
    yCoord.insert(y1);
    yCoord.insert(y2);
    events.push_back({ADD, x1, y1, y2});
    events.push_back({DEL, x2, y1, y2});
  }
  sort(events.begin(), events.end());
  events.push_back({ADD, 1000000001, 0, 0});

  int i = 0, S = 0, E = yCoord.size();
  for(int y : yCoord)
    tr[y] = i, rtr[i] = y, ++i;
  SegmentTree st(rtr.size());
  long long ans = 0, prevX = -1;
  for(auto [cmd, x, y1, y2] : events) {
    if(prevX != x) {
      ans += (x - prevX) * st.query(); 
      prevX = x;
    }
    st.update(1, S, E, tr[y1], tr[y2], cmd);
  }
  cout << ans;
}