[Swift] 프로그래머스 - 카카오 신규 아이디 추천

문제: https://programmers.co.kr/learn/courses/30/lessons/72410

코딩테스트 연습 - 신규 아이디 추천

 

1~7단계를 구현해주면 되는 문제였습니다.

import Foundation

func solution(_ new_id:String) -> String {
    // 1단계 new_id의 모든 대문자를 대응되는 소문자로 치환합니다.
    var result = new_id.lowercased()
    
    // 2단계 new_id에서 알파벳 소문자, 숫자, 빼기(-), 밑줄(_), 마침표(.)를 제외한 모든 문자를 제거합니다.
    let possible = "1234567890abcdefghijklmnopqrstuvwxyz-_."
    result = result.filter { possible.contains(String($0)) }
    
    // 3단계 new_id에서 마침표(.)가 2번 이상 연속된 부분을 하나의 마침표(.)로 치환합니다.
    var lastDot = -2
    var removeDot = [Int]()
    var resultArray = Array(result)
    
    if !resultArray.isEmpty {
        for i in 0..<result.count {
            if resultArray[i] == "." {
                if lastDot + 1 == i {
                    removeDot.append(i)
                }
                lastDot = i
            }
        }
    }
    
    if !removeDot.isEmpty {
        for i in 0..<removeDot.count {
            resultArray.remove(at: removeDot[i] - i)
        }
    }
    
    // 4단계 new_id에서 마침표(.)가 처음이나 끝에 위치한다면 제거합니다.
    if !resultArray.isEmpty {
        if resultArray.first == "." {
            resultArray.removeFirst()
        }
    }
    
    if !resultArray.isEmpty {
        if resultArray.last == "." {
            resultArray.removeLast()
        }
    }
    
    result = String(resultArray)
    
    // 5단계 new_id가 빈 문자열이라면, new_id에 "a"를 대입합니다.
    if result.count == 0 {
        result = "a"
    }
    
    // 6단계 new_id의 길이가 16자 이상이면, new_id의 첫 15개의 문자를 제외한 나머지 문자들을 모두 제거합니다.
    if result.count >= 16 {
        let resultArray = Array(result)
        result = String(resultArray[0...14])
        
        // 만약 제거 후 마침표(.)가 new_id의 끝에 위치한다면 끝에 위치한 마침표(.) 문자를 제거합니다.
        if result.last == "." {
            result.removeLast()
        }
    }
    
    // 7단계 new_id의 길이가 2자 이하라면, new_id의 마지막 문자를 new_id의 길이가 3이 될 때까지 반복해서 끝에 붙입니다.
    if result.count <= 2 {
        let last = result.last!
        
        while result.count < 3 {
            result.append(last)
        }
    }
    
    return result
}

solution("...!@BaT#*..y.abcdefghijklm") // "bat.y.abcdefghi"
solution("z-+.^.") // "z--"
solution("=.=") // "aaa"
solution("123_.def") // "123_.def"
solution("abcdefghijklmn.p") // "abcdefghijklmn"