11 ljedd2

[LJEDD2]

🔗 문제 링크

백준 우선순위 큐 | BOJ 2075 N번째 큰 수

문제 설명

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

이러한 표가 주어졌을 때, N번째 큰 수를 찾는 프로그램을 작성하시오.

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

✔️ 소요된 시간

1시간

✨ 수도 코드

우선순위큐 유형 문제 푸는데 재미가 들려 도전해본 문제입니다!
일단 각 줄에 주어진 수의 범위가 양 옆으로 10억이기 때문에 일반적인 접근으로는 어림도 없다는 것을 느낄 수 있었습니다.

따라서 N^2의 숫자들을 모두 살펴보면서 로직을 수행하되 이러한 입력들을 크기가 N 짜리 우선순위 큐에서 처리해주어야 합니다.

문제 풀이 과정

1. 우선순위 큐 안에 들어있는 원소의 개수가 N개 미만이라면 우선순위 큐에 집어넣습니다.
2. 우선순위 큐 안에 들어있는 원소의 개수가 N개라면   
   1. 현재 확인하고 있는 숫자가 우선순위 큐의 최솟값보다 크다면?      
      -> 우선순위 큐의 최솟값을 제거, 현재 확인하고 있는 숫자를 우선순위 큐에 삽입합니다.

위 과정을 풀어쓰면 아래와 같습니다

heappush()를 이용하면, 우선 순위에 따른 정렬이 일어나므로 찾는 값이 맨 앞으로 올 때 까지 맨 앞 값을 계속 제거해주고, N번째 큰 값을 출력할 수 있게 됩니다.

최종 코드

import sys, heapq
input = sys.stdin.readline

n = int(input())
heap = []

for _ in range(n):
    for number in map(int, input().split()):

        if len(heap) < n: # 비교 대상이 모자랄 경우 
            heapq.heappush(heap, number) #그대로 추가 

        else:
            if number > heap[0]: # 제일 작은것보다 크면 
                heapq.heapreplace(heap, number) #작은거 빼고 큰거 넣어줌 

print(heap[0]) # 맨앞에 있는게 N번째로 큰놈 

📚 새롭게 알게된 내용

[jung0115]

우선순위 큐 마스터로 가고 계시군요 🫢
저는 최근에 그리디 알고리즘 문제를 풀다가 우선순위 큐를 사용했는데, 처음에 완전 잘못 생각하고 있다가 우선순위 큐를 사용해야 한다는 걸 알고 어리둥절 했습니당...🥹 어떤 상황에서 우선순위 큐가 이용되면 좋을지 연습을 많이 해봐야 할 것 같아요!! 정은 님 덕분에 사용할 수 있는 곳 하나를 또 학습했습니다!

[janghw0126]

8차시 PR 때 이 문제를 한번 풀어보았는데 정은님이 푸셨던 로직과 거의 동일하네요! 저도 처음에는 N개의 수를 heap에 넣은 뒤에 다음 수부터는 N번째 큰 수보다 그 다음 행의 숫자가 더 크다면, heap에는 그 다음 행의 숫자를 대신 넣고 아니면 넣지 않는 방식으로 풀었습니다😊

이번 PR도 수고하셨습니다🌻

[janghw0126]

우와! 파이썬 heapq 모듈 중에서 heapreplace 함수라는 것이 있었군요!
정은님 덕분에 heapreplace 함수가 heap에서 최소값을 제거하고 item을 추가하는 역할을 한다는 것을 알아갑니다🫢
다음에 저도 한번 써보겠습니닷!