출처: 프로그래머스 코딩 테스트 연습

문제 설명

서비스에 가입하는 유저들의 아이디를 생성하는 업무를 담당하게 되었습니다. "네오"에게 주어진 첫 업무는 새로 가입하는 유저들이 카카오 아이디 규칙에 맞지 않는 아이디를 입력했을 때, 입력된 아이디와 유사하면서 규칙에 맞는 아이디를 추천해주는 프로그램을 개발하는 것입니다.
다음은 카카오 아이디의 규칙입니다.

아이디의 길이는 3자 이상 15자 이하여야 합니다.
아이디는 알파벳 소문자, 숫자, 빼기(-), 밑줄(_), 마침표(.) 문자만 사용할 수 있습니다.
단, 마침표(.)는 처음과 끝에 사용할 수 없으며 또한 연속으로 사용할 수 없습니다.
"네오"는 다음과 같이 7단계의 순차적인 처리 과정을 통해 신규 유저가 입력한 아이디가 카카오 아이디 규칙에 맞는 지 검사하고 규칙에 맞지 않은 경우 규칙에 맞는 새로운 아이디를 추천해 주려고 합니다.
신규 유저가 입력한 아이디가 new_id 라고 한다면,

1단계 newid의 모든 대문자를 대응되는 소문자로 치환합니다.
2단계 new_id에서 알파벳 소문자, 숫자, 빼기(-), 밑줄(), 마침표(.)를 제외한 모든 문자를 제거합니다.
3단계 new_id에서 마침표(.)가 2번 이상 연속된 부분을 하나의 마침표(.)로 치환합니다.
4단계 new_id에서 마침표(.)가 처음이나 끝에 위치한다면 제거합니다.
5단계 new_id가 빈 문자열이라면, new_id에 "a"를 대입합니다.
6단계 new_id의 길이가 16자 이상이면, new_id의 첫 15개의 문자를 제외한 나머지 문자들을 모두 제거합니다.
만약 제거 후 마침표(.)가 new_id의 끝에 위치한다면 끝에 위치한 마침표(.) 문자를 제거합니다.
7단계 new_id의 길이가 2자 이하라면, new_id의 마지막 문자를 new_id의 길이가 3이 될 때까지 반복해서 끝에 붙입니다.

예를 들어, new_id 값이 "...!@BaT#*..y.abcdefghijklm" 라면, 위 7단계를 거치고 나면 new_id는 아래와 같이 변경됩니다.

1단계 대문자 'B'와 'T'가 소문자 'b'와 't'로 바뀌었습니다.
"...!@BaT#*..y.abcdefghijklm" → "...!@bat#*..y.abcdefghijklm"

2단계 '!', '@', '#', '*' 문자가 제거되었습니다.

"...!@bat#*..y.abcdefghijklm" → "...bat..y.abcdefghijklm"

3단계 '...'와 '..' 가 '.'로 바뀌었습니다.
"...bat..y.abcdefghijklm" → ".bat.y.abcdefghijklm"

4단계 아이디의 처음에 위치한 '.'가 제거되었습니다.
".bat.y.abcdefghijklm" → "bat.y.abcdefghijklm"

5단계 아이디가 빈 문자열이 아니므로 변화가 없습니다.
"bat.y.abcdefghijklm" → "bat.y.abcdefghijklm"

6단계 아이디의 길이가 16자 이상이므로, 처음 15자를 제외한 나머지 문자들이 제거되었습니다.
"bat.y.abcdefghijklm" → "bat.y.abcdefghi"

7단계 아이디의 길이가 2자 이하가 아니므로 변화가 없습니다.
"bat.y.abcdefghi" → "bat.y.abcdefghi"

따라서 신규 유저가 입력한 new_id가 "...!@BaT#*..y.abcdefghijklm"일 때, 네오의 프로그램이 추천하는 새로운 아이디는 "bat.y.abcdefghi" 입니다.

문제

신규 유저가 입력한 아이디를 나타내는 new_id가 매개변수로 주어질 때, "네오"가 설계한 7단계의 처리 과정을 거친 후의 추천 아이디를 return 하도록 solution 함수를 완성해 주세요.

제한 사항

newid는 길이 1 이상 1,000 이하인 문자열입니다.
new_id는 알파벳 대문자, 알파벳 소문자, 숫자, 특수문자로 구성되어 있습니다.
new_id에 나타날 수 있는 특수문자는 -.~!@#$%^&*()=+[{]}:?,<>/ 로 한정됩니다.

입출력 예

no	new_id	result
예1	"...!@BaT#*..y.abcdefghijklm"	"bat.y.abcdefghi"
예2	"z-+.^."	"z--"
예3	"=.="	"aaa"
예4	"123_.def"	"123_.def"
예5	"abcdefghijklmn.p"	"abcdefghijklmn"

입출력 예 설명

입출력 예 #1
문제의 예시와 같습니다.

입출력 예 #2
7단계를 거치는 동안 new_id가 변화하는 과정은 아래와 같습니다.

1단계 변화 없습니다.
2단계 "z-+.^." → "z-.."
3단계 "z-.." → "z-."
4단계 "z-." → "z-"
5단계 변화 없습니다.
6단계 변화 없습니다.
7단계 "z-" → "z--"

입출력 예 #3
7단계를 거치는 동안 new_id가 변화하는 과정은 아래와 같습니다.

1단계 변화 없습니다.
2단계 "=.=" → "."
3단계 변화 없습니다.
4단계 "." → "" (new_id가 빈 문자열이 되었습니다.)
5단계 "" → "a"
6단계 변화 없습니다.
7단계 "a" → "aaa"

입출력 예 #4
1단계에서 7단계까지 거치는 동안 newid("123.def")는 변하지 않습니다. 즉, new_id가 처음부터 카카오의 아이디 규칙에 맞습니다.

입출력 예 #5
1단계 변화 없습니다.
2단계 변화 없습니다.
3단계 변화 없습니다.
4단계 변화 없습니다.
5단계 변화 없습니다.
6단계 "abcdefghijklmn.p" → "abcdefghijklmn." → "abcdefghijklmn"
7단계 변화 없습니다.

내가 제출한 코드

import Foundation

func solution(_ new_id:String) -> String {
    var result: String = new_id
    
    //step1
    result = new_id.lowercased()
    print("step1: \(result)")
    
    // step2
    result = result.filter {
        return ($0.isLetter || $0.isNumber || $0 == "-" || $0 == "_" || $0 == ".") ? true : false 
    }
    // step2 print
    print("step2: \(result)")
    
    // step3
    var tmpChr: Character = "."
    result = result.filter {
        if $0 == "." && tmpChr == $0{
            return false
        } else {
            tmpChr = $0
            return true
        }
    }
    print("step3: \(result)")
    
    //step4
    if result.prefix(1) == "." {
       result.removeFirst()
    }
    if result.suffix(1) == "." {
        result.removeLast()
    }
    
    print("step4: \(result)")
    
    //step5
    if result.count==0{
        result += "a"
    }
    print("step5: \(result)")
    
    //step6
    if result.count >= 16 {
        let lastIndex = result.index(result.startIndex, offsetBy: 14)
        result = String(result[...lastIndex])
        if result[lastIndex] == "." {
            result = String(result[..<lastIndex])
        }
    }
    print("step6: \(result)")
    
    // step7
    if result.count <= 2 {
        var lastString = String(result.suffix(1))
        for _ in 0..<(3-result.count) {
            result.append(lastString)
        }
    }
    print("step7: \(result)")
    return result
}

코드 설명

문제는 문제 설명에 기술된 대로로 7단계로 풀이를 진행한다.
각 단계마다 사용되는 문자열 메소드가 다르므로 각 단계별로 코드를 설명한다.

1단계 부터 살펴본다.

result = new_id.lowercased()

입력된 문자열을 소문자로 치환하는 과정이다. 문자열을 치환하는 메소드는 lowercased 메소드를 사용한다. 반대로 대문자로 치환하는 메소드는 uppercased 사용하면 된다. 2개의 메소드 모두 character, string에 적용이 가능하다.

2단계

result = result.filter {
        return ($0.isLetter || $0.isNumber || $0 == "-" || $0 == "_" || $0 == ".") ? true : false 
    }

소문자, 숫자, 빼기(-), 밑줄(_), 마침표(.)를 제외한 모든 문자를 제거하는 단계이다. 입력된 문자열의 각 문자에 접근하기 위해 고차 함수인 filter 메소드를 사용하였다. 알파벳 소문자인지 판별은 isLetter 메소드를 사용하였다. 숫자 판별은 isNumber 메소드를 사용하고, 다른 조건은 비교 연산자를 사용하였다.

3단계

var tmpChr: Character = "."
    result = result.filter {
        if $0 == "." && tmpChr == $0{
            return false
        } else {
            tmpChr = $0
            return true
        }
    }

연속된 마침표(.)를 제거하는 과정이다. 2단계와 마찬가지로 각 문자에 접근하기 위해 filter메소드를 사용하였다. filter 메소드가 문자열의 각 문자를 참조하는데, 이때 참조하는 문자에서 true를 반환하면 해당 문자를 새로운 문자열에 포함시키고 false를 반환할 시 해당 문자를 포함시키지 않는다. 마침표의 연속 여부를 확인하기 위해 현재 문자와 이전 문자를 비교하는 방법을 사용하였다. 이전 문자를 저장할 변수로 tmpChr를 선언하였고, 이전 문자와 현재 문자가 같으면 현재 문자를 제거하는 방식으로 진행한다.

4단계

//step4
    if result.prefix(1) == "." {
       result.removeFirst()
    }
    if result.suffix(1) == "." {
        result.removeLast()
    }

첫 문자와 마지막 문자를 확인하여 마침표(.)인 경우 제거하는 과정이다. 우선 처음과 마지막 문자를 확인하기 위해 prefix, suffix 메소드를 사용하였다. 그 후, 제거하는 과정은 removeFirst, removeLast 메소드를 사용한다. 참고로 prefix나 suffix에 사용하는 Int 값은 문자의 위치를 의미한다.(1일시 첫번째 문자를 가져옴)

5단계

if result.count==0{
        result += "a"
    }

빈 문자열인 경우 "a"를 추가하는 단계이다. 빈 문자열은 문자열 길이 프로퍼티인 count를 사용하였는데 이 대신 isEmpty 메소드를 사용해도 무방하다.

6단계

if result.count >= 16 {
        let lastIndex = result.index(result.startIndex, offsetBy: 14)
        result = String(result[...lastIndex])
        if result[lastIndex] == "." {
            result = String(result[..<lastIndex])
        }
    }

문자열의 길이가 16 이상이면 15까지 자르고, 마지막 문자가 마침표(.)인 경우 제거하는 단계이다. 문자열을 자르기 위해 String.index를 사용하였다. offsetBy에 할당하는 값은 문자열의 인덱스를 기준으로 인식하므로 14를 할당하였기 때문에 정확히 15번째 문자를 가르키게 된다. 여기서 문자열의 subscript를 이용한 접근은 subString 타입을 반환하므로 String타입으로 변환한다.

7단계

if result.count <= 2 {
        var lastString = String(result.suffix(1))
        for _ in 0..<(3-result.count) {
            result.append(lastString)
        }
    }

마지막 단계는 길이가 2이하인 경우 마지막 문자를 추가하는 과정이다. 마지막 문자는 suffix 메소드를 사용하여 얻었다. 이 메소드는 반환형이 Substring 타입이므로 마찬가지로 String타입으로 형변환을 한다.

몰랐던 사실 or 기억하면 도움이 될 만한 사실

Swift의 filter

알고리즘 문제를 풀 때, map과 filter를 자주 사용하는 것 같다. filter는 collection 예를들어 배열인 경우 각 element에 접근하거나 문자열에서 사용하는 경우 각 문자에 접근할 수 있다. filter의 declaration는 다음과 같다.

func filter(_ isIncluded: (Character) throws -> Bool) rethrows -> String

우선 filter 메소드 내부에서 Bool값 여부에 따라 새로 반환하는 문자열에 해당 문자가 포함이 될지 안될지 여부를 결정한다. 이는 문자열과 문자열 배열에서도 다음과 같이 사용이 가능하다.

let cast = ["Vivien", "Marlon", "Kim", "Karl"]
let shortNames = cast.filter { $0.count < 5 }
print(shortNames)
// Prints "["Kim", "Karl"]"

나머지 메소드들은 apple developer 문서를 참조하여 이해하면 된다.

https://developer.apple.com/documentation/swift/character/3127015-isnumber
https://developer.apple.com/documentation/swift/string/2894830-prefix
https://developer.apple.com/documentation/swift/string/2892530-suffix
https://developer.apple.com/documentation/swift/string/2892519-removefirst
https://developer.apple.com/documentation/swift/string/2892866-removelast
https://developer.apple.com/documentation/swift/character/3127011-isletter
https://developer.apple.com/documentation/swift/string/1641392-lowercased