백준 1655 - 가운데를 말해요 [Python]

www.acmicpc.net/problem/1655

1655번: 가운데를 말해요

 

중앙값을 기준으로 우선순위 큐 두 개를 만들어주고 풀었다.

 

코드 더러워지는거 싫어서

hpp = heapq.heappop

hps = heapq.heappush

라고 미리 써두고 풀었다

 

import heapq, sys
input = sys.stdin.readline
hpp = heapq.heappop
hps = heapq.heappush
l = []
r = []
n = int(input())
for i in range(n):
    t = int(input())
    if i < 2:
        hps(r, t)
        now = r[0]
        print(r[0])
        continue
    if t <= now:
        hps(l, -t)
    else:
        hps(r, t)
 
    if len(l) == len(r):
        lpop = hpp(l)
        hps(r, -lpop)
    if len(l) + 3 == len(r):
        rpop = hpp(r)
        hps(l, -rpop)
    now = r[0]
    print(r[0])