[백준 파이썬 2587번]대표값★우선순위 힙으로 풀어보기

2587번 대표

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

2587번: 대표값2

import sys
import heapq

res= []
for i in range(5):
    n = int(sys.stdin.readline())

    heapq.heappush(res , (n,i+1))
avg = 0
median = 0
j = 0
while res:
    a = heapq.heappop(res)
    print(a)
    j+=1
    if  j==3:
        median  = a[0]
    avg+=a[0]

print(avg//5)
print(median)

힙_활용2

==> 힙 pop은 heap의 [0]번째 인덱스가 작은 값을 먼저 출력한다.