코딩테스트

백준 11650 좌표 정렬하기(자바) - 정렬(Comparator)

leeeehhjj 2021. 9. 30. 16:04

좌표 정렬하기

시간 제한메모리 제한제출정답맞은 사람정답 비율

1 초 256 MB 54860 26449 20098 48.134%

문제

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

 


이 문제는 2차원 배열에 x좌표와 y 좌표를 입력한 후 x 좌표끼리 비교 후 정렬하고, x 가 같은 경우 y를 기준으로 정렬해야 한다. 그냥 Arrays.sort(arr)을 사용해서 2차원 배열을 정렬하는 것은 불가능하다.

따라서 이 경우에는 Comparator 인터페이스를 사용해야 한다.

 


import java.util.Arrays;
import java.util.Comparator;
import java.util.Scanner;

public class Main {
    public static void main(String args[]) {
        Scanner scanner = new Scanner(System.in);
        int n = scanner.nextInt();
        int[][] arr = new int[n][2];
        for (int i = 0; i < n; i++) {
            arr[i][0] = scanner.nextInt();
            arr[i][1] = scanner.nextInt();
        }

        Arrays.sort(arr, new Comparator<int[]>() {
            @Override
            public int compare(int[] e1, int[] e2) {
                if (e1[0] == e2[0]) {
                    return e1[1] - e2[1];
                }
                else {
                    return e1[0] - e2[0];
                }
            }
        });

        /* 람다식 표현현
       Arrays.sort(arr, (e1, e2) -> {
            if (e1[0] == e2[0]) {
                return e1[1] - e2[1];
            }
            else {
                return e1[0] - e2[0];
            }
        });*/

        for (int i = 0; i < n; i++) {
            System.out.println(arr[i][0] + " " + arr[i][1]);
        }
    }
}