[프로그래머스] 문자열 내 마음대로 정렬하기 (Kotlin)

문제 링크 : https://school.programmers.co.kr/learn/courses/30/lessons/12915

프로그래머스

 

문제 설명
- 문자열로 구성된 리스트 strings와, 정수 n이 주어졌을 때, 각 문자열의 인덱스 n번째 글자를 기준으로 오름차순 정렬하려 합니다.

- 예를 들어 strings가 ["sun", "bed", "car"]이고 n이 1이면 각 단어의 인덱스 1의 문자 "u", "e", "a"로 strings를 정렬합니다.

 

 

풀이

- 주어진 n 을 통해 compare 대상 문자열의 지정한 알파벳을 조건에 따라 비교한 이후, 단어를 비교해주면 된다.

- sortedWith {} 를 통해 조건에 따른 비교 정렬한다.

- compareTo를 통해 알파벳 순서대로 정렬 가능하다.

class Solution {
    fun solution(strings: Array<String>, n: Int): Array<String> {
        return strings.sortedWith { a, b ->
            if (a[n] == (b[n])) a.compareTo(b)
            else a[n].compareTo(b[n])
        }.toTypedArray()
    }
}