백준 / 11650번 / 좌표 정렬하기

[문제]

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

[입력]

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

[출력]

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

[예제 입력]

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

[예제 출력]

1 -1
1 1
2 2
3 3
3 4

 

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

public class Main {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);

        int n = sc.nextInt();
        int[][] arr = new int[n][2];

        sc.nextLine();
        for(int i=0; i<n; i++) {
            arr[i][0] = sc.nextInt();
            arr[i][1] = sc.nextInt();
        }

        Arrays.sort(arr, new Comparator<int[]>() {
            @Override
            public int compare(int[] arr1, int[] arr2) {
                if(arr1[0] == arr2[0]) {
                    return arr1[1]-arr2[1];
                } 
                return arr1[0]-arr2[0];
            }
        });
        
        for (int[] i : arr) {
            System.out.println(i[0] + " " + i[1]);
        }
    }
}

Arrays.sort를 사용하는데, Comparator 인터페이스를 사용헀다.

int형 2차원 배열을 선언하고, 값을 저장한 후

좌표를 x좌표가 증가하는 순으로 정렬하기 위해

return arr1[0] - arr2[0]로 x좌표를 비교한다.

 

x좌표가 같은 경우는 y좌표가 증가하는 순서로 정렬한다 (arr1[1] - arr2[1])

특정 규칙으로 정렬하는 Arrays.sort에 대해선 저번 포스팅에서 다뤘다.

 

11650번과 비슷한 문제로 11651번 (좌표 정렬하기 2) 문제 또한 비슷한 방식으로 풀 수 있다.

 

Arrays.sort Comparator 정렬 관련 포스팅 : https://gr616.tistory.com/279?category=945998