[BOJ] 백준 2075번 N번째 큰 수 - 파이썬(Python)

2075번: N번째 큰 수

문제

N×N의 표에 수 N2개 채워져 있다. 채워진 수에는 한 가지 특징이 있는데, 모든 수는 자신의 한 칸 위에 있는 수보다 크다는 것이다. N=5일 때의 예를 보자.

12	7	9	15	5
13	8	11	19	6
21	10	26	31	16
48	14	28	35	25
52	20	32	41	49

이러한 표가 주어졌을 때, N번째 큰 수를 찾는 프로그램을 작성하시오. 표에 채워진 수는 모두 다르다.

입력

첫째 줄에 N(1 ≤ N ≤ 1,500)이 주어진다. 다음 N개의 줄에는 각 줄마다 N개의 수가 주어진다. 표에 적힌 수는 -10억보다 크거나 같고, 10억보다 작거나 같은 정수이다.

출력

첫째 줄에 N번째 큰 수를 출력한다.

 

풀이

이 문제는 heapq(또는 우선순위 큐)를 사용하여 전체 중에서 N번째 큰 수를 찾아내면 되는 문제이다.

단, 이 문제의 경우 주어지는 메모리 크기가 작기 때문에 주어지는 모든 수를 heapq에 넣을 수 없기에 처음에는 N 개의 수를 heapq에 넣은 뒤 다음 수 부터는 heapq에서 N번째 큰 수와 비교하면서 크면 heapq에서 N번째 수 대신 집어 넣고 아니면 집어 넣지 않는 방식으로 풀어야 한다.

 

코드

import heapq

N = int(input())


heap = []

for i in range(N):
    nums = list(map(int, input().split()))
    if i == 0:
        for num in nums:
            heapq.heappush(heap, num)
    else:
        for num in nums:
            if num > heap[0]:
                heapq.heappop(heap)
                heapq.heappush(heap, num)



print(heapq.heappop(heap))