[백준] 11650 : 좌표 정렬하기 (Comparator<int[]> 이차원배열 정렬) - JAVA

백준 단계별 문제풀이 13단계 (정렬)

https://www.acmicpc.net/problem/11650

11650번: 좌표 정렬하기

 

 

# 문제

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

 

# 예제

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

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

 

출력

1 -1
1 1
2 2
3 3
3 4

 

# 필요개념

 

1차원 배열 정렬은 많이 해봤으나, 2차원 배열 정렬은 안 해봤다! 어차피 배열을 정렬하는 것이니 Arrays.sort를 사용할 거 같아 메소드를 조사해보았다.

 

✅ sort(T[] arr, Comparator<? super T> c)

지정된 비교기에 의해 유발된 순서에 따라 지정된 개체 배열을 정렬한다

 

sort의 인수는 2개였다. 첫번째는 정렬할 배열, 두번째 인수가 위 레퍼런스에서 말하는 '지정된 비교기' 가 되는 것이다. 

기본적으로 객체는 Comparable이 구현되어있는데, 구현되지 않은 클래스의 경우에는 셀프로 구현할 수 있다고 한다.

우리가 int형이나 String형의 일차원 배열을 오름차순 정렬할 경우에는 별도의 구현이 필요하지 않아 sort(arr)로 작성이 가능했으나, 2차원배열이나 내림차순 정렬을 할 땐 별도로 Comparator를 인수로 넣어줘야한다.

 

+) 내림차순

내림차순정렬을 할 땐 두번째 인수 자리에 Collections.reverseOrder()을 넣어줘야한다. 이는 Comparator 객체이다!

물론 내림차순 Comparator를 구현하는 것도 가능하다.

// Comparator로 구현
Arrays.sort(arr, new Comparator<Integer>() {
    @Override
    public int compare(Integer i1, Integer i2) {
        return i2 - i1;
    }
});

// 람다로 구현
Arrays.sort(arr, (i1, i2) -> i2 - i1);

 

 

✅ Comparator와 Comparable

 

Comparator와 Comparable는 인터페이스라서, 이 안에 선언된 메소드를 꼭! 구현해서 사용해야한다. 둘의 구현해야 할 메소드를 살펴보면 다음과 같다.

 

• Comparable의 compareTo(T o) 메소드
• Comparator의 compare(T o1, T o2)

두 인터페이스 모두 객체를 비교할 수 있게 하여 정렬해준다. compareTo는 1개의 인수를 받는다. 즉 자기자신과 인수 객체를 비교하는 메소드이다. 반면 compare은 두 개의 인수를 받아 두 객체를 비교한다. 한마디로 둘의 비교대상에 차이가 있다는 것이다!

 

우리는 해당 문제에서 자기자신과의 비교가 아닌 두 인수에 대한 비교를 하게 될 것이므로, Comparator을 쓸 것이다.

Arrays.sort의 두번째 인수에 Comparator을 생성해주면 된다! 이때 java.util을 import 해주어야 한다.

 

 

# Code

import java.io.*;
import java.util.*;

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[][] arr = new int[n][2];

        for (int i = 0 ; i < n ; i++) {
            StringTokenizer st = new StringTokenizer(br.readLine());
            arr[i][0] = Integer.parseInt(st.nextToken());
            arr[i][1] = Integer.parseInt(st.nextToken());
        }
        Arrays.sort(arr, new Comparator<int[]>() {
            @Override
            public int compare(int[] o1, int[] o2) {
                if (o1[0] == o2[0]) {
                    return o1[1] - o2[1];
                } else {
                    return o1[0] - o2[0];
                }
            }
        });
        for (int k = 0 ; k < n ; k++) System.out.println(arr[k][0] + " " + arr[k][1]);
    }
}

 

# 결과