[백준/Java] 11650 좌표 정렬하기(Comparator)

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

📝 풀이 3 : 이차원 배열 정렬(thenComparing())

`Comparator` 인터페이스의 `comparing()`, `thenComparing()` 메서드 사용⭐

import java.io.*;
import java.util.*;

public class Main {
    // 2차원 평면 위의 점 N개가 주어질 때, x좌표 -> y좌표 순서로 정렬 후 출력
    public static void main(String[] args) throws IOException {

        /* [1] 입력 */
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int N = Integer.parseInt(br.readLine()); // 2차원 평면 위 점의 개수 N (1 ≤ N ≤ 100,000)
        int[][] arrList = new int[N][2];
        for (int i = 0; i < N; i++) {
            StringTokenizer st = new StringTokenizer(br.readLine());
            arrList[i][0] = Integer.parseInt(st.nextToken());
            arrList[i][1] = Integer.parseInt(st.nextToken());
        }

        /* [2] 정렬 */
        Arrays.sort(arrList, Comparator.comparingInt((int[] a) -> a[0]).thenComparingInt(a -> a[1]));
        // (int[] a) -> a[0] 이 람다식은 배열 arr의 각 요소(즉, int[])를 받아서 그 배열의 첫 번째 요소(a[0])를 반환합니다. 따라서 arr의 요소들은 a[0]에 따라 정렬됩니다.
        // .thenComparingInt(a -> a[1]): 이 부분은 추가적인 정렬 기준을 지정합니다.
        // thenComparingInt 메서드는 첫 번째 정렬 기준(즉, a[0])이 동일한 경우에 두 번째 기준(즉, a[1])을 사용하여 정렬을 수행합니다.
        // a -> a[1] 이 람다식은 배열 a를 받아서 그 배열의 두 번째 요소(a[1])를 반환합니다. 여기서 a의 타입은 생략되었지만, 자바의 타입 추론에 의해 int[]로 인식됩니다.

        /* [3] 출력 */
        StringBuilder sb = new StringBuilder();
        for (int i = 0; i < N; i++) {
            sb.append(arrList[i][0] + " " + arrList[i][1]).append('\n');
        }
        System.out.println(sb);
    }
}

📝 풀이 2 : 이차원 배열 정렬(람다식)

람다식을 사용하여 Comparator 구현 코드 간결하게 작성⭐

import java.io.*;
import java.util.*;

public class Main {
    // 2차원 평면 위의 점 N개가 주어질 때, x좌표 -> y좌표 순서로 정렬 후 출력
    public static void main(String[] args) throws IOException {

        /* [1] 입력 */
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int N = Integer.parseInt(br.readLine()); // 2차원 평면 위 점의 개수 N (1 ≤ N ≤ 100,000)
        int[][] arrList = new int[N][2];
        for (int i = 0; i < N; i++) {
            StringTokenizer st = new StringTokenizer(br.readLine());
            arrList[i][0] = Integer.parseInt(st.nextToken());
            arrList[i][1] = Integer.parseInt(st.nextToken());
        }

        /* [2] 정렬 */
        Arrays.sort(arrList, (a, b) -> a[0] != b[0] ? a[0] - b[0] : a[1] - b[1]); // x좌표가 다를 경우 x좌표 오름차순

        /* [3] 출력 */
        StringBuilder sb = new StringBuilder();
        for (int i = 0; i < N; i++) {
            sb.append(arrList[i][0] + " " + arrList[i][1]).append('\n');
        }
        System.out.println(sb);
    }
}

📝 풀이 1 : Map<Integer, PriorityQueue>

`PriorityQueue` 이용⭐

import java.io.*;
import java.util.*;

public class Main {
    // 2차원 평면 위의 점 N개가 주어질 때, x좌표 -> y좌표 순서로 정렬 후 출력
    public static void main(String[] args) throws IOException {

        /* [1] 입력 */
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int N = Integer.parseInt(br.readLine()); // 2차원 평면 위 점의 개수 N (1 ≤ N ≤ 100,000)
        Map<Integer, PriorityQueue<Integer>> map = new TreeMap<>();
        for (int i = 0; i < N; i++) {
            String[] xy = br.readLine().split(" "); // i번점의 위치 xi와 yi(-100,000 ≤ xi, yi ≤ 100,000)
            int x = Integer.parseInt(xy[0]);
            map.computeIfAbsent(x, k -> new PriorityQueue<>()).add(Integer.parseInt(xy[1]));
            // x좌표에 해당하는 PriorityQueue가 없으면 새로 생성(이미 존재하면 기존 값을 반환)하고 y좌표 추가
        }

        /* [2] 출력 */
        StringBuilder sb = new StringBuilder();
        for (Map.Entry<Integer, PriorityQueue<Integer>> entry : map.entrySet()) {
            int x = entry.getKey();
            PriorityQueue<Integer> yQueue = entry.getValue();
            while (!yQueue.isEmpty()) {
                sb.append(x + " " + yQueue.poll() + "\n"); // for-each 문으로 순회할 때는 요소들이 정렬된 순서로 반환되지 않음
            }
        }
        System.out.println(sb);
    }
}