ps:problems:boj:18436
수열과 쿼리 37
ps | |
---|---|
링크 | acmicpc.net/… |
출처 | BOJ |
문제 번호 | 18436 |
문제명 | 수열과 쿼리 37 |
레벨 | 골드 1 |
분류 |
구간 쿼리 |
시간복잡도 | O(n+mlogn) |
인풋사이즈 | n<=100,000, m<=100,000 |
사용한 언어 | Python |
제출기록 | 44656KB / 692ms |
최고기록 | 664ms |
해결날짜 | 2021/03/24 |
풀이
- A 수열에 대응되는 B 수열이 있어서, A[i]가 홀수이면 B[i]=1, 짝수이면 B[i]=0 이라고 하자.
- A[l:r]중에서 홀수의 갯수를 구하는 것은 sum(B[l:r])과 같고, 짝수의 갯수는 (r-l+1)-sum(B[l:r]) 과 같다.
- 결국 포인트 업데이트가 있는 구간 합 문제이므로, 펜윅트리를 사용해서 각 쿼리를 O(logn)에 처리할 수 있다.
- 펜윅트리 구축에 O(n), m개의 쿼리를 각 O(logn)에 처리하는 데에 O(mlogn). 도합 O(n+mlogn)에 해결 가능하다
코드
"""Solution code for "BOJ 18436. 수열과 쿼리 37".
- Problem link: https://www.acmicpc.net/problem/18436
- Solution link: http://www.teferi.net/ps/problems/boj/18436
"""
import sys
from teflib import fenwicktree
def main():
N = int(sys.stdin.readline()) # pylint: disable=unused-variable
A = [int(x) for x in sys.stdin.readline().split()]
M = int(sys.stdin.readline())
fenwick = fenwicktree.FenwickTree(x % 2 for x in A)
for _ in range(M):
query = [int(x) for x in sys.stdin.readline().split()]
if query[0] == 1:
_, i, x = query
fenwick.set(i - 1, x % 2)
elif query[0] == 2:
_, l, r = query
print(r - l + 1 - fenwick.query(l - 1, r))
else:
_, l, r = query
print(fenwick.query(l - 1, r))
if __name__ == '__main__':
main()
- Dependency: teflib.fenwicktree.FenwickTree
ps/problems/boj/18436.txt · 마지막으로 수정됨: 2021/04/30 16:08 저자 teferi
토론