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

문제

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

 

해설

최소 힙을 사용하여 해결하는 문제이다. 공간복잡도를 고려해야하는 문제이므로 모든 데이터를 입력받은 후 pop을 진행하는 것이 아닌 input 과정 중 필요한 공간만큼만 사용하는 것이 핵심이었다.

 

# PyPy3
import heapq


def solution(cnt):

    heap = []
    for _ in range(cnt):
        for v in list(map(int, input().split())):
            heapq.heappush(heap, v)
            if len(heap) == cnt+1:
                heapq.heappop(heap)

    return heapq.heappop(heap)


n = int(input())
print(solution(n))