ps:problems:boj:12895
화려한 마을
ps | |
---|---|
링크 | acmicpc.net/… |
출처 | BOJ |
문제 번호 | 12895 |
문제명 | 화려한 마을 |
레벨 | 플래티넘 3 |
분류 |
구간 쿼리 |
시간복잡도 | O(n+mtlogn) |
인풋사이즈 | n<=100,000, t<=30, m<=100,000 |
사용한 언어 | Python |
제출기록 | 37740KB / 5720ms |
최고기록 | 5084ms |
해결날짜 | 2021/03/27 |
풀이
- lazy propagation를 이용해서 푸는 구간 업데이트 + 구간 쿼리 문제.
- 원래 구간에서 서로 다른 수의 갯수를 찾는 것은 기본적으로 업데이트가 있을 때에는 효율적으로 처리하기 어려운 종류의 쿼리이다.
- 그러나 이 문제에서는 사용할 색의 갯수가 매우 작기 때문에, 각각의 노드에 그 노드가 관리하는 구간 내에 존재하는 색의 목록을 모두 저장하고 있는 것이 가능하다. 이렇게 하면 구간을 합치는 것도 O(T)에 처리 가능하고, 색 목록을 bitmask를 사용해서 저장한다면 단순한 or연산이면 된다.
- 업데이트 연산은 단순한 assignment 연산이 된다.
- lazy propagation에서 사용한 노테이션대로, f,g,h 를 정의하면 아래처럼 된다.
- f(x, y) = x|y.
- g(x, c, size) = c
- h(c1, c2) = c2
- 구간 쿼리를 해서 f값을 얻은 다음, 그 안에 켜져있는 비트 수를 세면 된다.
- 세그먼트 트리를 구축하는 데에 O(n), m개의 쿼리를 각각 O(t*logn)에 처리하는 데에 O(m*t*logn), 합쳐서 O(n+m*t*logn)이다
코드
"""Solution code for "BOJ 12895. 화려한 마을".
- Problem link: https://www.acmicpc.net/problem/12844
- Solution link: http://www.teferi.net/ps/problems/boj/12844
"""
import operator
import sys
from teflib import segmenttree
def main():
N, T, Q = [int(x) for x in sys.stdin.readline().split()]
lazy_segtree = segmenttree.LazySegmentTree(
[1] * N,
merge=operator.or_,
update_value=lambda value, param, size: param,
update_param=lambda param1, param2 : param2)
for _ in range(Q):
query = sys.stdin.readline().split()
if query[0] == 'C':
x, y, z = [int(x) for x in query[1:]]
lazy_segtree.range_update(x - 1, y, 1 << (z - 1))
else:
x, y = [int(x) for x in query[1:]]
ret = lazy_segtree.query(x - 1, y)
print(bin(ret).count('1'))
if __name__ == '__main__':
main()
- Dependency: teflib.segmenttree.LazySegmentTree
ps/problems/boj/12895.txt · 마지막으로 수정됨: 2021/04/30 15:13 저자 teferi
토론