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

코딩테스트 연습 - 문자열 내 마음대로 정렬하기

 

문제 설명

1. 문자열로 구성된 리스트와 정수가 주어졌을 때, 각 문자열의 n번째 글자를 기준으로 오름차순 정렬

2. n번째 글자가 같을 경우 사전순으로 앞선 문자열이 앞에 위치

 

 

풀이

1. 하라는 대로 구현하면 됩니다 ..

2. swift의 정렬 함수를 공부할 수 있는 문제

 

 

코드

func solution(_ strings:[String], _ n:Int) -> [String] {
    var ans: [String] = strings
    ans.sort(by: {
        let s1: String = $0
        let index1 = s1.index(s1.startIndex, offsetBy: n)
        let s2: String = $1
        let index2 = s2.index(s2.startIndex, offsetBy: n)
        if(s1[index1] == s2[index2]) {
            return s1 < s2
        }
        return s1[index1] < s2[index2]
    })
    return ans
}

 

 

 

코드 개선

func solution(_ strings:[String], _ n:Int) -> [String] {
    return strings.sorted(by: {
        let s1: String = $0
        let index1 = s1.index(s1.startIndex, offsetBy: n)
        let s2: String = $1
        let index2 = s2.index(s2.startIndex, offsetBy: n)
        if(s1[index1] == s2[index2]) {
            return s1 < s2
        }
        return s1[index1] < s2[index2]
    })
}