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

programmers.co.kr - 코딩테스트연습 - Lv.1 - 연습문제 - 문자열 내 마음대로 정렬하기

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

Constraints :

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

 

Solution.swift :

//
//  Created by Yongwoo Marco on 2022/05/01.
//  Copyright © 2022 Yongwoo Marco Kim. All rights reserved.
//

func solution(_ strings:[String], _ n:Int) -> [String] {
    return strings.sorted { lhs, rhs in
        let lhsChar = lhs[lhs.index(lhs.startIndex, offsetBy: n)]
        let rhsChar = rhs[rhs.index(rhs.startIndex, offsetBy: n)]
        return lhsChar == rhsChar ? lhs < rhs : lhsChar < rhsChar
    }
}

print(solution(["sun", "bed", "car"], 1)) // ["car", "bed", "sun"]
print(solution(["abce", "abcd", "cdx"], 2)) // ["abcd", "abce", "cdx"]

How I tried this :
주어진 인덱스 번째 문자를 비교해서 같으면 문자 자체를 비교,

다르면 해당 문자의 오름차순으로 정렬하는 정렬 코드를 작성

 

문제에 관한 모든 저작권 : https://programmers.co.kr/