[프로그래머스] 문자열 내 마음대로 정렬하기 자바 list.sort Comparator

https://school.programmers.co.kr/learn/courses/30/lessons/12915

프로그래머스

문제

문자열로 구성된 배열에서 주어진 인덱스 n을 기준으로 각 단어별로 n번째 글자를 기준으로 오름차순으로 정렬하여 반환하시오.

 

제한사항

인덱스 n의 문자가 같은 문자열이 여럿 일 경우, 사전순으로 앞선 문자열이 앞쪽에 위치합니다.

접근

list의 sort(Comparator.comparing) 을 활용해 특정 인덱스를 기준으로 정렬하게끔 작성하였다.

public String[] solution(String[] strings, int n) {
        List<String> list = new ArrayList<>(Arrays.asList(strings));
        list.sort(Comparator.comparing(a -> a.charAt(n)));
        return (new String[0]);
    }

발견된 문제

제한사항을 준수하지 못함.
-> Comparator 체이닝을 활용할 수 있다고 한다. 이를 적용하면 아래와 같다.

public String[] solution2(String[] strings, int n) {
        List<String> list = new ArrayList<>(Arrays.asList(strings));

        list.sort(Comparator
                .comparing((String a) -> a.charAt(n))  // 1. n번째 문자 기준 정렬
                .thenComparing(Comparator.naturalOrder())  // 2. 사전순 정렬 (동률 시)
        );

        return list.toArray(new String[0]);
    }

알게된 점

1. 특정 문자열을 기준으로 정렬하기 (List)
list.sort(Comparator.comparing( (String a) -> a.charAt( idx ) ) )
1.1 정렬 후 재정렬 (ex 사전순 정렬)
list.sort(Comparator.comparing( (String a) -> a.charAt( idx ) )
         .thenComparing(Comparator.naturalOrder()
);