[JAVA] 기본 정렬 대신 원하는 방식으로 정렬하기

Comparator와 Comparable

정렬을 하기 위해서는 기본적으로 ①대상 ②기준  두가지가 필요하다. Arrays와 Collections 클래스의 sort() 메서드는 대상과 정렬 기준을 인자로 전달하면 반환값 없이 내부에서 정렬한다.

 

//    import java.util.Arrays

Integer[] intArr = {3, 6, 2, 1, 4, 5};
Arrays.sort(intArr);    //    intArr = [1, 2, 3, 4, 5, 6]

 

 

이때 정렬 기준 없이 대상만 전달할 경우 각 클래스에 구현되어있는 기본 정렬 기준으로 정렬되는데, Arrays와 Collections는 (숫자와 알파벳처럼 순서로 비교가 가능한 값이라면) 오름차순으로 정렬할 수 있도록 구현되어있다.

 

여기서 정렬 기준으로 사용된 것은 Comparable이다. 이 인터페이스에 있는 compare() 메서드는 오름차순으로 정렬할 수 있도록 구현되어있다. Arrays와 Collections은 둘 다 Comparable의 구현체이다.

 

 

다음은 프로그래머스의 [Lv. 1][문자열 내 마음대로 정렬하기] 문제다.

 

 

문제 설명

문자열로 구성된 리스트 `strings`와, 정수 `n`이 주어졌을 때, 각 문자열의 인덱스 n번째 글자를 기준으로 오름차순 정렬하려 합니다. 예를 들어 `strings`가 `["sun", "bed", "car"]`이고 n이 1이면 각 단어의 인덱스 1의 문자 `"u", "e", "a"`로 `strings`를 정렬합니다.

 

 

import java.util.*;

class Solution {
    public String[] solution(String[] strings, int n) {
        String[] answer = strings.clone();

        Arrays.sort(answer, new Comparator<String>() {
            public int compare(String str1, String str2) {
//                같으면 기본 정렬
                if (str1.charAt(n) == str2.charAt(n)) {
                    return str1.compareTo(str2);
                }

//                n번째 인덱스로 정렬
                return str1.charAt(n) - str2.charAt(n);
            }
        });

        return answer;
    }
}

 

 

 

Array.sort()를 호출하면서 대상인 answer을 전달하고, n번째 문자로 정렬하는 기준을 새로 정의하여 인자로 전달하기 위해 compare() 메서드를 오버라이딩 했다. 어차피 인자로 쓰이고 버려지기 때문에 별도의 참조 변수 선언 없이 객체를 생성했다.

같은 내용으로 람다식을 적용할 수 있다.

 

 

Arrays.sort(answer2, (String str1, String str2) -> {
//          같으면 기본 정렬
            if (str1.charAt(n) == str2.charAt(n)) {
                return str1.compareTo(str2);
            }

//          n번째 인덱스로 정렬
            return str1.charAt(n) - str2.charAt(n);
        }
    );