[Java] 백준 2075 (N번째 큰 수) 자바 문제 풀이

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

2075번: N번째 큰 수

 

알고리즘 분류

• 자료 구조
• 정렬
• 우선순위 큐

풀이

정렬을 해서 풀어야 하는 문제이다. 하지만 메모리 제한이 12MB이므로 2차원 배열을 다 입력받고 정렬을 했다간 메모리 초과가 날 것 같아, 우선 순위 큐로 문제를 접근했다.

 

PriorityQueue를 선언하고 람다식을 사용해 ( (o1, o2) -> o2 - o1 ) 높은 수 부터 정렬했다.

PriorityQueue<Integer> queue = new PriorityQueue<>( (o1, o2) -> o2 - o1 );
String [] temp;
  for (int i = 0; i < n; i++) {
       temp = br.readLine().split(" ");
       for (int j = 0; j < temp.length; j++) {
            queue.add(Integer.parseInt(temp[j]));
       }
}

 

 

아래는 전체 정답 코드이다.

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.PriorityQueue;

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int n = Integer.parseInt(br.readLine());

        PriorityQueue<Integer> queue = new PriorityQueue<>( (o1, o2) -> o2 - o1 );
        String [] temp;
        for (int i = 0; i < n; i++) {
            temp = br.readLine().split(" ");
            for (int j = 0; j < temp.length; j++) {
                queue.add(Integer.parseInt(temp[j]));
            }
        }
        int answer = 0;
        int cnt = 0;
        while (cnt < n ){
            answer = queue.poll();
            cnt++;
        }
        System.out.println(answer);

    }
}