[프로그래머스 코딩테스트] 문자열 내 마음대로 정렬하기(Java)

  문제 설명

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

 

  제한사항

• strings는 길이 1 이상, 50 이하인 배열입니다.
• strings의 원소는 소문자 알파벳으로 이루어져 있습니다.
• strings의 원소는 길이 1 이상, 100 이하인 문자열입니다.
• 모든 strings의 원소의 길이는 n보다 큽니다.
• 인덱스 1의 문자가 같은 문자열이 여럿 일 경우, 사전순으로 앞선 문자열이 앞쪽에 위치합니다.

 

  입출력 예

strings	n	return
["sun", "bed", "car"]	1	["car", "bed", "sun"]
["abce", "abcd", "cdx"]	2	["abcd", "abce", "cdx"]

import java.util.Arrays;
class Solution {
    public String[] solution(String[] strings, int n) {
        Arrays.sort(strings, (a, b) -> {
            if (a.charAt(n) == b.charAt(n)) {
                return a.compareTo(b);
            } else {
                return Character.compare(a.charAt(n), b.charAt(n));
            }
        });
        String[] answer = Arrays.copyOf(strings, strings.length);
        return answer;
    }
}

  Arrays 클래소의 sort 메소드를 이용하여 정렬할 것입니다. 그러나 평소 정렬과는 달리 조건에 따라 정렬해야 하므로 람다식을 이용하여 정렬 기준을 지정합니다. strings 배열에서 a와 b를 기준으로 비교합니다.

  if문에서 a의 n번째 문자가 b의 n번째 문자와 동일하다면 compare 메소드를 이용하여 a와 b를 비교하여 오름차순으로 반환합니다.

  같지 않다면 Character 클래스의 compare 메소드를 이용하여 a의 n번째 문자와 b의 n번째 문자를 비교하여 반환합니다.

  answer 배열을 생성하고 Arrays 클래스의 copyOf 메소드를 이용하여 정렬된 배열 strings를 전부 복사하여 할당합니다.