https://www.acmicpc.net/problem/12895
12895번: 화려한 마을
첫 번째 줄에 N, T, Q (1 ≤ N ≤ 100,000, 1 ≤ T ≤ 30, 1 ≤ Q ≤ 100,000)이 공백을 구분으로 주어진다. 각각 천나라에 존재하는 집의 개수, 사용할 색의 개수, 작업의 개수를 의미한다. 두 번째 줄부터 작
www.acmicpc.net
출력은 l ~ r 구간에 칠해져 있는 색의 가짓수, 집에 칠할 수 있는 색은 최대 30가지입니다.
업데이트와 쿼리가 섞여 있으므로 오프라인 쿼리 및 mo's 알고리즘은 사용하지 못하고, 세그먼트 트리 lazy를 사용합니다.
트리에는 해당 구간에 칠해져 있는 색의 가짓수를 비트 필드로 저장합니다.
우선 모든 집에 1이 칠해져 있는 상태로 시작합니다. init 함수에서 초기화를 진행합니다.
type == C면 l ~ r 구간에 업데이트를 진행합니다.
원래 칠해져 있던 색을 지우고, 새로운 색을 채워야 하므로 tree[node]에 해당 색으로 변경해줍니다.
type == Q면 l ~ r 구간에 칠해져 있는 색의 가짓수를 출력합니다.
트리에는 비트 필드를 저장했으니 시프트연산을 하면서 1의 갯수를 출력합니다.
lazyUpdate는 모든 update, query 시 먼저 진행합니다.
이 문제는 l > r인 입력도 주어지니 l < r이 되도록 swap을 해야합니다.
|
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
|
#include <iostream>
#define MAX 100001
using namespace std;
int tree[MAX * 4], lazy[MAX * 4];
int N, M, K;
void init(int node, int s, int e) {
if (s == e) {
tree[node] = 1;
return;
}
int m = (s + e) / 2;
init(node * 2, s, m);
init(node * 2 + 1, m + 1, e);
tree[node] = tree[node * 2] | tree[node * 2 + 1];
}
void lazyUpdate(int node, int s, int e) {
if (!lazy[node]) return;
tree[node] = lazy[node];
if (s != e) {
lazy[node * 2] = lazy[node];
lazy[node * 2 + 1] = lazy[node];
}
lazy[node] = 0;
}
void update(int node, int s, int e, int l, int r, int k) {
lazyUpdate(node, s, e);
if (s > r || l > e) return;
if (l <= s && e <= r) {
lazy[node] = (1 << k);
lazyUpdate(node, s, e);
return;
}
int m = (s + e) / 2;
update(node * 2, s, m, l, r, k);
update(node * 2 + 1, m + 1, e, l, r, k);
tree[node] = tree[node * 2] | tree[node * 2 + 1];
}
int query(int node, int s, int e, int l, int r) {
lazyUpdate(node, s, e);
if (s > r || l > e) return 0;
if (l <= s && e <= r) return tree[node];
int m = (s + e) / 2;
return query(node * 2, s, m, l, r) | query(node * 2 + 1, m + 1, e, l, r);
}
void func() {
char type;
int l, r, k;
while (K--) {
cin >> type;
if (type == 'C') {
cin >> l >> r >> k;
if (l > r) swap(l, r);
update(1, 1, N, l, r, k - 1);
}
else {
cin >> l >> r;
if (l > r) swap(l, r);
int ret = query(1, 1, N, l, r);
int ans = 0;
while (ret) {
ans += (ret & 1);
ret >>= 1;
}
cout << ans << '\n';
}
}
}
void input() {
cin >> N >> M >> K;
init(1, 1, N);
}
int main() {
cin.tie(NULL); cout.tie(NULL);
ios::sync_with_stdio(false);
input();
func();
return 0;
}
|
cs |
'algorithm > SegmentTree' 카테고리의 다른 글
| boj 1517 버블 소트 (0) | 2024.06.16 |
|---|---|
| boj 17131 여우가 정보섬에 올라온 이유 (3) | 2024.04.25 |
| boj 10167 금광 (0) | 2022.09.02 |
| boj 16993 연속합과 쿼리 (0) | 2022.08.29 |
| boj 12846 무서운 아르바이트 (0) | 2022.08.19 |
