[BOJ] N번째 큰 수 2075번 C++

문제

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번째 큰 수를 출력한다.

 

접근 방식

 

우선순위 큐를 사용하면 간단하게 풀 수 있을 거라 생각했다

모든 값들을 최대 힙에 넣어주고 4번 pop을 해주어서 N번째 큰 수를 출력하려 했더니 메모리 초과가 발생하였다

 

따라서 다른 분들 풀이를 살펴보니

최소 힙을 사용해 풀이하는 방법이 있었다

모든 값들을 똑같이 최소 힙에 넣어주되 만약 최소 힙의 사이즈가 N 이상이면 pop을 진행해준다. 이 때 가장 작은 최솟값이 pop되어짐

 

그렇게 하면 N번째 숫자가 최소 힙에 top에 오게된다!

 

 

코드

#include<bits/stdc++.h>
using namespace std;
int N;
priority_queue<int, vector<int>, greater<>> pq;

int main() {
    ios::sync_with_stdio(false); cin.tie(0);
    cin>>N;
    int x;
    for(int i=0; i<N*N; i++) {
        cin>>x;
        pq.push(x);
        if(N < (int)pq.size()) pq.pop();
    }
    cout<<pq.top();
}