백준 2075 N번째 큰 수 Python

🐸 문제 정보

2075번: N번째 큰 수

 

🤖 알고리즘

우선순위 큐

 

⏱️ 풀이 시간

22.33m

 

📝 풀이

구현처럼 풀다가 실패해서 검색으로 아이디어를 얻었다.

우선순위 큐로 푸는 문제였고,  n*n이기 때문에 큐의 범위를 n만큼 유지한다면 마지막 줄까지 돌았을 때 0번째에 위치한 원소가 n번째 큰 수가 되었다.

 

오늘은 집중이 잘 안되는 날이다ㄷㄷ

 

🧑‍💻 나의 답

# pypy3

import sys, heapq
input = sys.stdin.readline

n = int(input().rstrip())
q = []

for _ in range(n):
    nums = list(map(int, input().rstrip().split()))
    for num in nums:
        if len(q) < n:
            heapq.heappush(q, num)
        else:
            if q[0] < num:
                heapq.heappop(q)
                heapq.heappush(q, num)

print(q[0])