[Algorithm] 백준 1927 : 최소 힙

채멈·2024년 1월 25일

Algorithm

목록 보기
13/24
post-thumbnail

문제
https://www.acmicpc.net/problem/1927
풀이
https://github.com/nowChae/algorithm/blob/master/%EB%B0%B1%EC%A4%80/Silver/1927.%E2%80%85%EC%B5%9C%EC%86%8C%E2%80%85%ED%9E%99/%EC%B5%9C%EC%86%8C%E2%80%85%ED%9E%99.py

힙 관련 뼈대 문제로 heapq 모듈을 사용하여 쉽게 풀어주었다. heapq 모듈을 사용하면 최소힙을 기본으로 하기때문에 heapq 라이브러리의 메소드를 사용하여 바로 풀어 주었다.

< 풀이 코드 >

import heapq
import sys
input = sys.stdin.readline

N = int(input())
heap = []

for _ in range(N):
  i = int(input())

  if i == 0:
    if len(heap) == 0:
      print(0)
    else:
      print(heapq.heappop(heap))
  else:
    heapq.heappush(heap, i)
profile
공부 기록 차곡차곡 ( ੭ ・ᴗ・ )੭

0개의 댓글