백준 2075 : N번째 큰 수

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

2075번: N번째 큰 수

[ 문제풀이 ]

구하고자 하는 것이 N번째로 큰 수이기 때문에 최소 힙 형태의 우선순위 큐를 활용하여 N개의 큰 수를 관리해주면 된다.

우선순위 큐 크기가 N보다 작을 때는 수를 삽입해주고, N개일 때부터는 현재 수가 최솟값보다 큰 경우 갱신해주면 된다.

import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.util.PriorityQueue;
import java.util.StringTokenizer;

public class Main {

	public static void main(String[] args) {
		try (BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
				BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out))) {
			int N = Integer.parseInt(br.readLine());
			PriorityQueue<Integer> pq = new PriorityQueue<>();
			for (int i = 0; i < N; i++) {
				StringTokenizer st = new StringTokenizer(br.readLine());
				for (int j = 0; j < N; j++) {
					int num = Integer.parseInt(st.nextToken());
					if (pq.size() <= N - 1) {
						pq.offer(num);
						continue;
					}
					
					if (pq.peek() < num) {
						pq.poll();
						pq.offer(num);
					}
				}
			}
			bw.write(pq.poll() + "\n");
		} catch (Exception e) {
			e.printStackTrace();
		}
	}
}