ABOUT ME

-

Today
-
Yesterday
-
Total
-
  • 백준 11650 좌표 정렬하기(자바) - 정렬(Comparator)
    코딩테스트 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]);
            }
        }
    }
Designed by Tistory.