11286

날짜 : 2023. 02. 15

사용 언어 : python

 

문제

 

 

코드

import sys
import heapq

n = int(sys.stdin.readline())
q = []

for i in range(n):
    x = int(sys.stdin.readline()) # 값 입력
    if x != 0:
        heapq.heappush(q, (abs(x), x)) # 절댓값 기준으로 heapify, 쌍으로 저장
    else:
        if len(q) == 0: # 힙이 비어있는 경우
            print(0)
        else: # 비어있지 않은 경우
            print(heapq.heappop(q)[1]) # 절댓값에 대응하는 값 출력

 

 

풀이

- 입력 시 (절댓값, 원본값) 쌍을 이루는 튜플을 만든 다음 절댓값을 기준으로 min heap을 구현한다.

- 이후 절댓값에 따라 heap pop을 통해 값을 출력한다. 출력하는 값을 튜플의 두 번째에 위치한 원본을 출력

 

 

알게된 점

- 단순 heap push를 사용했을 때 리스트가 자동적으로 정렬된다는 것으로 잘못알고 있었다.

- heap은 트리에서의 max, min을 출력하기 위해 사용되는 것이기 때문에 리스트는 일정부분만 정렬되었는데 완전히 오개념으로 인지했었다.

 

 

참고 사이트

- https://hellominchan.tistory.com/231

[Python] heapq 사용법