알고리즘 - 프로그래머스 문자열 다루기 기본 (Swift)

 

문자열 다루기 기본

문자열 s의 길이가 4 혹은 6이고, 숫자로만 구성돼있는지 확인해주는 함수, solution을 완성하세요. 예를 들어 s가 a234이면 False를 리턴하고 1234라면 True를 리턴하면 됩니다.

제한 사항

• s는 길이 1 이상, 길이 8 이하인 문자열입니다.

---

코드를 짤때 항상 지금은 이조건이지만 추후 확장성을 생각하면서 짜다보니 코드양이 길어지고 불필요한 로직이 많이 들어가는 것 같다.

간략하게는 인트타입으로 형변환해서 nil체크 이후 갯수만 확인해서 리턴해도 될 것 같다.

func solution(_ s:String) -> Bool {
    let numString = "1234567890"
    var isString = true
    let filterNumString = s.filter {
        if numString.contains($0) {
            return true
        } else {
            isString = false
            return false
        }
    }

    return isString ? filterNumString.count == 4 || filterNumString.count == 6 : false
}