[Baekjoon] 11650 좌표 정렬하기 - JAVA

문제

2차원 평면 위의 점 N개가 주어진다. 좌표를 x좌표가 증가하는 순으로, x좌표가 같으면 y좌표가 증가하는 순서로 정렬한 다음 출력하는 프로그램을 작성하시오.

입력

첫째 줄에 점의 개수 N (1 ≤ N ≤ 100,000)이 주어진다. 둘째 줄부터 N개의 줄에는 i번점의 위치 xi와 yi가 주어진다. (-100,000 ≤ xi, yi ≤ 100,000) 좌표는 항상 정수이고, 위치가 같은 두 점은 없다.

출력

첫째 줄부터 N개의 줄에 점을 정렬한 결과를 출력한다.

예제 입력 1

5
3 4
1 1
1 -1
2 2
3 3

예제 출력 1

1 -1
1 1
2 2
3 3
3 4

 

---

✏️ 제출 답안

import java.io.*;
import java.util.Arrays;
import java.util.Comparator;
import java.util.StringTokenizer;

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());

        int[][] matrix = new int[n][2];

        for (int i = 0; i < n; i++) {
            StringTokenizer st = new StringTokenizer(br.readLine());
            matrix[i][0] = Integer.parseInt(st.nextToken());
            matrix[i][1] = Integer.parseInt(st.nextToken());
        }

        Arrays.sort(matrix, new Comparator<int[]>() {
            @Override
            public int compare(int[] o1, int[] o2) {
                return o1[0]==o2[0] ? o1[1]-o2[1] : o1[0]-o2[0];
            }
        });

        for (int i = 0; i < n; i++) {
            StringBuilder sb = new StringBuilder();
            sb.append(matrix[i][0]).append(" ").append(matrix[i][1]);
            System.out.println(sb.toString());
        }
    }
}

 

이번 문제는 앞에 풀었던 문제들과 유사한 정렬 문제여서 Comparator를 활용하면 된다는 생각이 바로 들긴 했지만, 내가 생각하지 못했던 부분은 삼항연산자로 쓴 부분이었다. 0번 열을 먼저 비교해서 값이 같으면 1번 열을 기준으로 정렬하고, 기 외의 경우는 0번 열을 기준으로 정렬되도록 했어야 했다. 처음에는 단순히 o1[0] - o2[0] 로 0번 열에 대해서만 정렬을 해서 제대로 정렬되지 않아 틀렸었다. 

 

제출 현황