[python] 백준 2075 - N번째 큰 수

Q> N x N 의 표에 수가 N^2개 채워져 있을 때

(모든 수는 자신 한 칸 위의 수보다 큼)

N번째 큰 수 출력하기

 

🧨 처음에 다 넣고 시도하니까

메모리 초과, 시간 초과 난다

N개만 가지고 넣었다뺐다 해주면 된다 !

 

import sys
import heapq
input = sys.stdin.readline

n = int(input().rstrip())  # n개의 수 받아오기
heap = []

for x in map(int,input().split()):  # 처음 n개만 저장
    heapq.heappush(heap,x)

for y in range(1,n):  # 그 뒤 부터는
  for z in map(int,input().split()):
    heapq.heappush(heap,z)  # 넣고
    heapq.heappop(heap)  # 가장 작은거 빼고

print(heapq.heappop(heap)) # 남은 것들 중 가장 작은것