알고리즘 공부/기타

[백준] 11279번 최대 힙- 파이썬

코딩 코딩 코오딩 2022. 3. 3. 14:46

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

 

11279번: 최대 힙

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

www.acmicpc.net

https://www.daleseo.com/python-heapq/

 

[파이썬] heapq 모듈 사용법

Engineering Blog by Dale Seo

www.daleseo.com

위의 방식만 사용하면 쉽다

힙 잘알아두자

import sys
import heapq

num = int(sys.stdin.readline().rstrip())
num_list = []

for i in range(num):
    x = int(sys.stdin.readline().rstrip())
    if x != 0:
        heapq.heappush(num_list,(-x,x))
    elif x == 0:
        if num_list:
            answer = heapq.heappop(num_list)
            print(answer[1])
        else :
            print(0)
반응형