[프로그래머스] Swift - 문자열 내 마음대로 정렬하기 (LV.1)

👆문제풀러 가기👆

문제 설명

문자열로 구성된 리스트 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"]

 

---

풀이 [ 메모리: 16.5mb, 최대시간: 1.57ms ] 

func solution(_ strings:[String], _ n:Int) -> [String] {
    var strings = strings.map { $0.map { String($0) } }
    
    strings.sort {
        if $0[n] == $1[n] {
            return $0.joined() < $1.joined()
        }
        return $0[n] < $1[n]
    }
    
    return strings.map { $0.joined() }
}

코드를 짜는게 쉬웠는데, 문제 이해가 잘 안 되었던 문제였다.

'인덱스 1의 문자가 같은 문자열이 여럿 일 경우, 사전순으로 앞선 문자열이 앞쪽에 위치합니다.'라는 조건에서 글자별 인덱스를 하나씩 뒤로 넘기며 사전순 오름차순을 해야한다고 생각했는데, 그게 아니라 단순히 문자 전체에 대한 비교였다.

문제를 이해한 뒤에는 바로 풀린 문제..