[Swift] 프로그래머스 | 코딩테스트 연습 -> 2021 KAKAO BLIND RECRUITMENT -> 신규 아이디 추천

func solution(_ new_id:String) -> String {
    
    func stage1(_ str: String) -> String {
    // 1단계 new_id의 모든 대문자를 대응되는 소문자로 치환합니다.
        return str.lowercased()
    }
    func stage2(_ str: String) -> String {
    // 2단계 new_id에서 알파벳 소문자, 숫자, 빼기(-), 밑줄(_), 마침표(.)를 제외한 모든 문자를 제거합니다.
        var res = str
        for character in res {
            switch character {
            case "a", "b", "c", "d", "e", "f", "g", "h", "i", "j", "k", "l", "m", "n", "o", "p", "q", "r", "s", "t", "u", "v", "w", "x", "y", "z", "0", "1", "2", "3", "4", "5", "6", "7", "8", "9", "-", "_", ".":
                break;
            default:
                res = res.replacingOccurrences(of: String(character), with: "")
            }
        }
        return res
    }
    func stage3(_ str: String) -> String {
    // 3단계 new_id에서 마침표(.)가 2번 이상 연속된 부분을 하나의 마침표(.)로 치환합니다.
        var res = str
        while res.contains("..") {
            res = res.replacingOccurrences(of: "..", with: ".")
        }
        
        return res
    }
    func stage4(_ str: String) -> String {
    // 4단계 new_id에서 마침표(.)가 처음이나 끝에 위치한다면 제거합니다.
        var res = str
        while ((res.first == ".") || (res.last == ".")) {
            if (res.first == "."){
                 res.removeFirst()
             }else if (res.last == "."){
                 res.removeLast()
             }
        }
        return res
    }
    func stage5(_ str: String) -> String {
    // 5단계 new_id가 빈 문자열이라면, new_id에 "a"를 대입합니다.
        if (str == ""){
            return "a"
        }
        return str
    }
    func stage6(_ str: String) -> String {
    // 6단계 new_id의 길이가 16자 이상이면, new_id의 첫 15개의 문자를 제외한 나머지 문자들을 모두 제거합니다.
    // 만약 제거 후 마침표(.)가 new_id의 끝에 위치한다면 끝에 위치한 마침표(.) 문자를 제거합니다.
        var res = str
        if (res.count >= 16){
            res.removeLast(res.count - 15)
            if (res.last == "."){
                res.removeLast()
            }
        }
        return res
    }
    func stage7(_ str: String) -> String {
    // 7단계 new_id의 길이가 2자 이하라면, new_id의 마지막 문자를 new_id의 길이가 3이 될 때까지 반복해서 끝에 붙입니다.
        var res = str
        if (res.count <= 2){
            while res.count < 3 {
                res.append(res.last!)
            }
        }
        return res
    }
    
    let one = stage1(new_id)
    let two = stage2(one)
    let three = stage3(two)
    let four = stage4(three)
    let five = stage5(four)
    let six = stage6(five)
    let seven = stage7(six)
    return seven
}

 

stage2에서 if-else문으로 작성하려 했는데,

알파벳 소문자, 숫자, 빼기(-), 밑줄(_), 마침표(.) 라는 조건이 매우 많고 이것은 하드코딩하여 스트링타입으로 만들어주고 예외처리를 해줘야 할 것 같아,

switch문을 이용하여 작성했다.

 

stage3에서 ".."을 "."으로 바꾸도록 하여도,

"."이 3번 연속되면, "..." -> ".." 이 되어 원하는 결과를 얻을수 없다. 따라서 while문으로 반복하도록 하였다.

 

또한, ".a."같은 경우, while문이 없다면, stage4에서 첫번째 "."만 제거되고, 결과는 a..이 될 것이다.

따라서 이것 또한 while문으로 반복하여 확인하도록 하였다.

 

예외상황들에 대해 더 꼼꼼히 생각하고 코드를 작성해야 한다.