기록하고 까먹지 말기

11286 본문

전공/백준

11286

yha97 2023. 2. 15. 21:50

날짜 : 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 사용법

[Python] heapq 사용법 (글쓴날 : 2020.04.26) * 이 글은 글쓴이가 공부한 내용을 정리하며 올리는 글입니다. * Python3를 기준으로 작성되었습니다. Python heapq 사용법 1) heapq란? heapq란 자료구조 queue의 일종

hellominchan.tistory.com

 

'전공 > 백준' 카테고리의 다른 글

1926  (0) 2023.02.23
1976  (0) 2023.02.22
2564  (0) 2023.02.15
7562  (0) 2023.02.14
2583  (0) 2023.02.09