Trulyawesome

[Swift] 문자열 내 마음대로 정렬하기 본문

알고리즘/프로그래머스

[Swift] 문자열 내 마음대로 정렬하기

ethkelitt 2020. 9. 10. 09:23

 

1
2
3
4
5
6
7
8
func solution(_ strings:[String], _ n:Int-> [String] {
    return strings.sorted {
        let a = $0[$0.index($0.startIndex, offsetBy: n)]
        let b = $1[$1.index($1.startIndex, offsetBy: n)]
        
        return first == second ? $0 < $1 : a < b
    }
}
cs

 

String의 index를 얻기위해 위와같이 공식문서에서 index함수를 찾아서 사용했습니다.