[BaekJoon] 11651번 좌표 정렬하기 2 문제풀이 (Success)

11651번: 좌표 정렬하기 2

 

☕Language: Java

 

import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Comparator;
import java.util.List;
import java.util.StringTokenizer;
 
public class Main {
 
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
 
        int num = Integer.parseInt(br.readLine());
 
        int[][] intArrXY = new int[num][2];
 
        for (int i = 0; i < num; i++) {
            StringTokenizer st = new StringTokenizer(br.readLine(), " ");
 
            int a = Integer.parseInt(st.nextToken());
            int b = Integer.parseInt(st.nextToken());
 
            intArrXY[i][0] = a;
            intArrXY[i][1] = b;
 
        }
 
        Arrays.sort(intArrXY, new Comparator<int[]>() {
            @Override
            public int compare(int[] o1, int[] o2) {
                return o1[1] != o2[1] ? o1[1] - o2[1] : o1[0] - o2[0];
            }
        });
 
        for (int i = 0; i < num; i++) {
            bw.write(intArrXY[i][0] + " " + intArrXY[i][1] + "\n");
        }
 
        bw.flush();
        bw.close();
    }
 
}
 

🤔 해설

1. sort()

    - 이중 배열의 경우, 기존 sort 메서드를 통해 정렬할 수 없으므로 정렬 기준을 오버라이딩하여 이차원 배열 정렬

2. @Override

    - Comparator 클래스에서 compare 메서드 Override

    - 배열의 두번째 요소가 다르면 두번째 요소로 정렬, 같으면 첫 번째 요소로 정렬

    - 예시

        - 입력값: (3, 4), (1, 1), (1, -1)

        - intArr[0][0] = 3, intArr[0][1] = 4

        - intArr[1][0] = 1, intArr[1][1] = 1

        - intArr[2][0] = 1, intArr[2][1] = -1

        - intArr = {{3, 4}, {1, 1}, {1, -1}}

        - int[] o1 = {3, 4}, int[] o2 = {1, 1}

        - int[0] o1 = 3, int[0] o2 = 1

        - int[0] o1 - int[0] o2 : 0번째 요소를 기준으로 오름 차순으로 정렬

 

😮 이 외의 풀이

import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Comparator;
import java.util.List;
import java.util.StringTokenizer;
 
public class Main {
 
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
 
        int num = Integer.parseInt(br.readLine());
 
        int[][] intArrXY = new int[num][2];
 
        for (int i = 0; i < num; i++) {
            StringTokenizer st = new StringTokenizer(br.readLine(), " ");
 
            int a = Integer.parseInt(st.nextToken());
            int b = Integer.parseInt(st.nextToken());
 
            intArrXY[i][0] = a;
            intArrXY[i][1] = b;
 
        }
 
        Arrays.sort(intArrXY, (arr1, arr2) -> {
            if (arr1[1] == arr2[1]) {
                return arr1[0] - arr2[0];
            } else {
                return arr1[1] - arr2[1];
            }
        });
 
        StringBuffer sb = new StringBuffer();
        for (int i = 0; i < num; i++) {
            sb.append(intArrXY[i][0] + " " + intArrXY[i][1] + "\n");
        }
 
        bw.write(sb + "");
        bw.flush();
        bw.close();
    }
 
}
 

1. 람다식(익명 클래스)를 Comparator 클래스 대신 사용

2. StringBuffer 사용

 

⭐ 메서드 사용 방법을 자세히 알고 싶으면, Eclipse 기준 ctrl + method 클릭

    public static <T> void sort(T[] a, Comparator<? super T> c) {
        if (c == null) {
            sort(a);
        } else {
            if (LegacyMergeSort.userRequested)
                legacyMergeSort(a, c);
            else
                TimSort.sort(a, 0, a.length, c, null, 0, 0);
        }
    }
 

- sort 사용 시, 첫 번째 인자가 T[]이므로 T는 int[]

 

 

 

🔗 소스 코드
HJ0216/TIL/BOJ