[9/19] 절댓값 힙 (x)

이경준·2021년 9월 19일
0

코테

목록 보기
120/140
post-custom-banner

실버1 문제 실패

코드

import heapq
import sys

n = int(sys.stdin.readline().rstrip())
arr = []

for _ in range(n):
    x = int(sys.stdin.readline().rstrip())
    
    if ( x != 0 ):
        heapq.heappush(arr, [abs(x), x])

    else:
        if ( len(arr) == 0 ):
            print(0)
        else:
            num = heapq.heappop(arr)[1]
            print(num)

로직

  • 힙에 절대값과 원래값을 같이 넣어준다.

피드백

  • 힙에 여러개 넣어도 됨
  • 힙은 첫번째 값을 기준으로 정렬된다.
  • try, except 문법 공부 필요
profile
The Show Must Go On
post-custom-banner

0개의 댓글