Algorithm Study/Python

[백준 파이썬] # 11279 최대 힙

728x90
반응형

Silver II

# 11279 최대 힙

링크 : https://www.acmicpc.net/problem/11279

 

11279번: 최대 힙

첫째 줄에 연산의 개수 N(1 ≤ N ≤ 100,000)이 주어진다. 다음 N개의 줄에는 연산에 대한 정보를 나타내는 정수 x가 주어진다. 만약 x가 자연수라면 배열에 x라는 값을 넣는(추가하는) 연산이고, x가

www.acmicpc.net

 

풀이

import heapq
import sys

N = int(sys.stdin.readline())
heap = []
for i in range(N):
    x = int(sys.stdin.readline())
    if x != 0:
        heapq.heappush(heap, -x)
    else:
        if len(heap) == 0:
            print(0)
        else:   
            pop = heapq.heappop(heap)
            print(abs(pop))

 

후기

  • 최대 힙을 사용하는 법에 대해 배울 수 있었다.

  • 파이썬에서 일반적으로 힙을 구현하면 최소 힙으로 구현된다. 그런데 heappush과정에서 - (마이너스)를 사용하는 트릭으로 최대 힙을 구현할 수 있다.

  • sys를 사용하지 않으면 시간 초과로 오답 처리된다.