[Programmers][Python][Level 1] 신규 아이디 추천

프로그래머스의 Level 01 2021 KAKAO BLIND 신규 아이디 추천 문제입니다.

 

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

 

문제 

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

 

• 아이디의 길이는 3자 이상 15자 이하여야 합니다.
• 아이디는 알파벳 소문자, 숫자, 빼기(-), 밑줄(_), 마침표(.) 문자만 사용할 수 있습니다.
• 단, 마침표(.)는 처음과 끝에 사용할 수 없으며 또한 연속으로 사용할 수 없습니다.

"네오"는 다음과 같이 7단계의 순차적인 처리 과정을 통해 신규 유저가 입력한 아이디가 카카오 아이디 규칙에 맞는 지 검사하고 규칙에 맞지 않은 경우 규칙에 맞는 새로운 아이디를 추천해 주려고 합니다.
신규 유저가 입력한 아이디가 new_id 라고 한다면,

 

• 1단계 new_id의 모든 대문자를 대응되는 소문자로 치환합니다.
• 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 함수를 완성해 주세요.

 

제한사항

• new_id는 길이 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단계까지 거치는 동안 new_id("123_.def")는 변하지 않습니다. 즉, new_id가 처음부터 카카오의 아이디 규칙에 맞습니다.

 

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

 

풀이

 

첫 번째 도전

테스트 코드 5개는 모두 통과를 하지만 제출을 하니 50점이 나오는군요...

왜? 50점이 나오지 생각을 해보다가 아래 16행의 3번 체크에서 이상한 점을 찾았습니다.

 

new_id가 만약 ...bat..y.abcdefghijklm ..이 맨 앞에 3개가 아니라 4개이거나 그 이상이라면 16행에서 제가 처리한 replace 두 번으로는 안되고 반복문을 통해서 해결을 해야 합니다.

import re

def solution(new_id):
    
    # new_id = ...!@BaT#*..y.abcdefghijklm
    
    # 1 : new_id 모든 대문자를 소문자로 치환
    new_id = new_id.lower()
    # print(new_id) : ...!@bat#*..y.abcdefghijklm
    
    # 2 : 소문자, 숫자, -, _, . 제외한 모든 문자 제거
    new_id = re.sub('[^a-z1-9-_.]', '', new_id)
    # print(new_id) : ...bat..y.abcdefghijklm
    
    # 3 : new_id에서 .가 2번 이상 연속된 부분을 하나의 .로 치환
    new_id = new_id.replace('..', '.').replace('..', '.')
    # print(new_id) : .bat.y.abcdefghijklm
    
    # 4 : 아이디의 처음과 끝에 위치한 .를 제거
    new_id = re.sub('^\.', '', new_id)
    # print(new_id) : bat.y.abcdefghijklm
    
    new_id = re.sub('\.$', '', new_id)
    # print(new_id) : bat.y.abcdefghijklm
    
    # 5 : new_id가 빈 문자열이면 new_id에 "a"를 대입
    if not new_id:
        new_id = "a"
    # print(new_id) : 조건 성립 x
    
    # 6 : new_id의 길이가 16자 이상이면, new_id의 첫 15개의 문자를 제외한 나머지 문자들을 모두 제거합니다.
    # 만약 제거 후 마침표(.)가 new_id의 끝에 위치한다면 끝에 위치한 마침표(.) 를 제거
    if len(new_id) >= 16:
        new_id = new_id[:15]
        # print(new_id) : bat.y.abcdefghi
        
    if new_id.endswith('.'):
        new_id = re.sub('\.$', '', new_id) 
        # print(new_id) : 조건 성립 x
    
    # 7 : new_id의 길이가 2자 이하라면 new_id의 마지막 문자를 new_id의 길이가 3이 될 때까지 반복해서 끝에 붙인다.
    if len(new_id) <= 2:
        while len(new_id) < 3:
            new_id += new_id[len(new_id)-1]
    
    # print(new_id) : bat.y.abcdefghi
    
    return new_id

 

50/100점

 

두 번째 도전

기존 15행을 find() 함수를 사용해서 ..이 -1이 아닌경우 즉 ..이 존재할때까지 ..을 .으로 치환해줍니다.

그리고 다시 채점을 해봅니다.

결과는 ...88.5점입니다... 흠 이번에는 다른 분들의 풀이를 보기로 했습니다.

import re

def solution(new_id):
    answer = ''
    
    # 1 : new_id 모든 대문자를 소문자로 치환
    new_id = new_id.lower()
    
    # 2 : 소문자, 숫자, -, _, . 제외한 모든 문자 제거
    new_id = re.sub('[^a-z1-9-_.]', '', new_id)
    
    # 3 : new_id에서 .가 2번 이상 연속된 부분을 하나의 .로 치환
    while new_id.find('..') != -1:
        new_id = new_id.replace('..', '.')
    
    # 4 : 아이디의 처음과 끝에 위치한 .를 제거
    new_id = re.sub('^\.', '', new_id)
    new_id = re.sub('\.$', '', new_id)
    
    # 5 : new_id가 빈 문자열이면 new_id에 "a"를 대입
    if not new_id:
        new_id = "a"
    
    # 6 : new_id의 길이가 16자 이상이면, new_id의 첫 15개의 문자를 제외한 나머지 문자들을 모두 제거합니다.
    # 만약 제거 후 마침표(.)가 new_id의 끝에 위치한다면 끝에 위치한 마침표(.) 를 제거
    if len(new_id) >= 16:
        new_id = new_id[:15]
    
    if new_id.endswith('.'):
        new_id = re.sub('\.$', '', new_id) 
    
    # 7 : new_id의 길이가 2자 이하라면 new_id의 마지막 문자를 new_id의 길이가 3이 될 때까지 반복해서 끝에 붙인다.
    if len(new_id) <= 2:
        while len(new_id) < 3:
            new_id += new_id[len(new_id)-1]
    
    
    return new_id

 

88.5/100점

 

 

세 번째 도전

다른 사람들의 풀이를 보다가 2번째 소문자, 숫자, (-, _, .)을 제외한 모든 문자를 제거하는 정규식에서 숫자를 1-9로 줘서 0이 포함되지 않아서 3개의 테스트케이스가 통과하지 않았습니다ㅜㅜ 역시 실수는 사소한 곳에서 발생하는 같습니다.

import re

def solution(new_id):
    
    # 1 : new_id 모든 대문자를 소문자로 치환
    new_id = new_id.lower()
    
    # 2 : 소문자, 숫자, -, _, . 제외한 모든 문자 제거
    new_id = re.sub('[^a-z0-9-_.]', '', new_id)
    
    # 3 : new_id에서 .가 2번 이상 연속된 부분을 하나의 .로 치환
    while new_id.find('..') != -1:
        new_id = new_id.replace('..', '.')
    
    # 4 : 아이디의 처음과 끝에 위치한 .를 제거
    new_id = re.sub('^\.', '', new_id)
    new_id = re.sub('\.$', '', new_id)
    
    # 5 : new_id가 빈 문자열이면 new_id에 "a"를 대입
    if not new_id:
        new_id = "a"
    
    # 6 : new_id의 길이가 16자 이상이면, new_id의 첫 15개의 문자를 제외한 나머지 문자들을 모두 제거합니다.
    # 만약 제거 후 마침표(.)가 new_id의 끝에 위치한다면 끝에 위치한 마침표(.) 를 제거
    if len(new_id) > 15:
        new_id = new_id[:15]
        if new_id[-1] == '.':
            new_id = new_id[:-1]
    
    # 7 : new_id의 길이가 2자 이하라면 new_id의 마지막 문자를 new_id의 길이가 3이 될 때까지 반복해서 끝에 붙인다.
    while len(new_id) < 3:
       new_id += new_id[-1]
        
    return new_id

 

 

다른 사람의 풀이

저는 주로 정규식을 이용해서 풀이를 했었지만, 다른 분은 인덱싱을 이용해서 해결을 하셨습니다.

def solution(new_id):
    answer = ''
    # 1
    new_id = new_id.lower()
    # 2
    for c in new_id:
        if c.isalpha() or c.isdigit() or c in ['-', '_', '.']:
            answer += c
    # 3
    while '..' in answer:
        answer = answer.replace('..', '.')
    # 4
    if answer[0] == '.':
        answer = answer[1:] if len(answer) > 1 else '.'
    if answer[-1] == '.':
        answer = answer[:-1]
    # 5
    if answer == '':
        answer = 'a'
    # 6
    if len(answer) > 15:
        answer = answer[:15]
        if answer[-1] == '.':
            answer = answer[:-1]
    # 7
    while len(answer) < 3:
        answer += answer[-1]
    return answer