이차원 배열 정렬하기(백준 10814)

 2차원 배열 정렬하는 방법

백준 10814번을 풀면서 2차원 배열을 정렬하는 문제가 나왔다.

💡 Arrays.sort와 Comparator를 활용한 배열 정렬

배열에는 아래와 같이 데이터가 들어있다.

arr[i][0]를 기준으로 정렬해야한다.

String[][] arr = {
    {"21", "Junkyu"},
    {"21", "Dohyun"},
    {"20", "Sunyoung"}
};

 

 

2차원 배열의 경우, 특정 열(column)을 기준을 정렬하려면 Comparator을 사용해야한다.

compare메소드는 두 개의 매개변수를 비교한다.

 

o1과 o2는 비교 대상 배열의 두 행을 나타낸다.

여기서는 두 행의 0번째 요소를 비교한다.

Arrays.sort(arr, new Comparator<String[]>() {
    @Override
    public int compare(String[] o1, String[] o2) {
        return Integer.parseInt(o1[0]) - Integer.parseInt(o2[0]); // 나이 비교
    }
});

 

 

 

java8이상부터는 이렇게 람다식으로 간단하게 작성할 수 있다.

Arrays.sort(arr, (o1, o2) -> Integer.parseInt(o1[0]) - Integer.parseInt(o2[0]));

 

 

🔍  Comparator

Comparator는 자바에서 객체의 정렬 기준을 지정할 때 사용하는 인터페이스

 

🔍 compare

compare(T o1, T o2)메서드는 두 객체 o1과 o2를 비교하고, 정렬 순서를 반환한다.

음수 : o1이 o2보다 작음(앞에 위치)

0     : o1과 o2이 같음

음수 : o1이 o2보다 큼(뒤에 위치)

 

🔍 compareTo

compareTo는 Comparable의 인터페이스 메서드로, 객체 자체에 기본 정렬 기준을 정의한다.

비교하고 싶다면 객체로 변환해서 비교해야한다.

 

 

 

 10814 : 나이순 정렬

문제

온라인 저지에 가입한 사람들의 나이와 이름이 가입한 순서대로 주어진다. 이때, 회원들을 나이가 증가하는 순으로, 나이가 같으면 먼저 가입한 사람이 앞에 오는 순서로 정렬하는 프로그램을 작성하시오.

 

입력

첫째 줄에 온라인 저지 회원의 수 N이 주어진다. (1 ≤ N ≤ 100,000)

둘째 줄부터 N개의 줄에는 각 회원의 나이와 이름이 공백으로 구분되어 주어진다. 나이는 1보다 크거나 같으며, 200보다 작거나 같은 정수이고, 이름은 알파벳 대소문자로 이루어져 있고, 길이가 100보다 작거나 같은 문자열이다. 입력은 가입한 순서로 주어진다.

 

출력

첫째 줄부터 총 N개의 줄에 걸쳐 온라인 저지 회원을 나이 순, 나이가 같으면 가입한 순으로 한 줄에 한 명씩 나이와 이름을 공백으로 구분해 출력한다.

입	출력
3 21 Junkyu 21 Dohyun 20 Sunyoung	20Sunyoung 21 Junkyu 21 Dohyun

 

실행 코드

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Arrays;
import java.util.Comparator;
import java.util.StringTokenizer;

public class P10814 {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st;

        int N = Integer.parseInt(br.readLine());
        String[][] arr = new String[N][2];

        for(int i=0;i<N;i++){
            st = new StringTokenizer(br.readLine());
            arr[i][0] = st.nextToken();
            arr[i][1] = st.nextToken();
        }
        Arrays.sort(arr, new Comparator<String[]>() {
            @Override
            public int compare(String[] o1, String[] o2) {
                return Integer.parseInt(o1[0]) - Integer.parseInt(o2[0]); //첫번째 기준 오름차순
            }
        });

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

 

 

백준 10814번을 풀면서 아래와 같이 정렬하는 문제가 있