파이썬) 백준 알고리즘 | 2075번 : N번째 큰 수

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

2075번: N번째 큰 수

백준 2075 'N번째 큰 수' 문제입니다.

그냥 정렬해서 풀면 메모리 문제가 나기 때문에 우선순위큐를 이용해야 합니다.

문제는 N by N이고 N번째 큰 수 를 원하기 때문에 우선순위큐에는 N개의 원소가 들어가게 하였습니다.

그래서 매 줄마다 크기를 비교하여 가장 큰 N개의 원소만 가지고 있게 구현하였습니다.

 

import sys
import heapq
input = sys.stdin.readline

N = int(input())
heap = []
for i in range(N):
    nums = list(map(int, input().split()))

    if not heap:
        for num in nums:
            heapq.heappush(heap, num)
    else:
        for num in nums:
            if heap[0] < num:
                heapq.heappop(heap)
                heapq.heappush(heap, num)
print(heap[0])