====== 수열과 쿼리 15 ======
===== 풀이 =====
* [[ps:구간 쿼리#구간 최솟값|업데이트 쿼리와 구간 최솟값 쿼리]]를 살짝 변형해서, 구간 최솟값의 인덱스를 물어보는 문제이다.
* 그냥 각 노드에 값 대신에, (값, 인덱스)의 튜플을 저장해주면 구간 최솟값을 위한 세그먼트 트리를 그대로 사용해서 풀수 있다.
* 하지만, 이 문제는 부분구간이 아니라 전체 범위에서의 최솟값만을 묻는데, 그렇다면 세그먼트 트리가 아니라 우선순위큐로도 해결이 가능하다.
* 사실, 세그먼트 트리보다 우선순위큐가 먼저 떠올라야 하는게 정상인거 같은데, '수열과 쿼리' 시리즈가 다 구간 쿼리만 다루다 보니까 오히려 잘 안떠올랐다. 문제에 태그도 세그먼트 트리만 걸려있다.
* teflib.priorityqueue.UpdatableHeap 을 써서 업데이트를 처리해도 되지만, 그냥 그 로직을 코드에 직접 구현한 이유는 그렇게 해도 심플하게 처리되어서이다.
* 시간 복잡도는 어느쪽으로 하든 O(n+mlogn)이지만, [[#코드 1 - 우선순위큐|우선순위큐]] 버전이 [[#코드 2 - 세그먼트 트리|세그먼트 트리]]버전보다 3배가량 빨랐다.
===== 코드 =====
==== 코드 1 - 우선순위큐 ====
* 51624KB / 468ms
"""Solution code for "BOJ 14427. 수열과 쿼리 15".
- Problem link: https://www.acmicpc.net/problem/14427
- Solution link: http://www.teferi.net/ps/problems/boj/14427
"""
import sys
import heapq
def main():
N = int(sys.stdin.readline())
A = [int(x) for x in sys.stdin.readline().split()]
heap = [(a_i, i + 1) for i, a_i in enumerate(A)]
heapq.heapify(heap)
M = int(sys.stdin.readline())
for i in range(M):
query = [int(x) for x in sys.stdin.readline().split()]
if query[0] == 1:
_, i, v = query
heapq.heappush(heap, (v, i))
A[i - 1] = v
while heap[0][0] != A[heap[0][1] - 1]:
heapq.heappop(heap)
else:
print(heap[0][1])
if __name__ == '__main__':
main()
==== 코드 2 - 세그먼트 트리 ====
* 50312KB / 1384ms
"""Solution code for "BOJ 14427. 수열과 쿼리 15".
- Problem link: https://www.acmicpc.net/problem/14427
- Solution link: http://www.teferi.net/ps/problems/boj/14427
"""
import sys
from teflib import segmenttree
def main():
N = int(sys.stdin.readline())
A = [int(x) for x in sys.stdin.readline().split()]
segtree = segmenttree.SegmentTree((a_i, i + 1) for i, a_i in enumerate(A))
M = int(sys.stdin.readline())
for i in range(M):
query = [int(x) for x in sys.stdin.readline().split()]
if query[0] == 1:
_, i, v = query
segtree.set(i - 1, (v, i))
else:
print(segtree.query(0, N)[1])
if __name__ == '__main__':
main()
* Dependency: [[:ps:teflib:segmenttree#SegmentTree|teflib.segmenttree.SegmentTree]]
{{tag>BOJ ps:problems:boj:골드_1}}