[python] 44. 신규 아이디 추천

문제 출처- kakao bline recruitment 2021

programmers.co.kr/learn/courses/30/lessons/72410

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

문제 설명

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

 

  - 아이디의 길이는 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가 매개변수로 주어질 때, "네오"가 설계한 7단계의 처리 과정을 거친 후의 추천 아이디를 return 하도록 solution 함수를 완성해 주세요.

 

제한사항

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

---

나의 연습장

1. 대문자를 소문자로 치환 👉🏻lower()함수

2. 알파벳, 숫자, 빼기, 밑줄, 마침표를 제외한 문자 제거 👉🏻re모듈의 compile, findall 함수로 정규표현식 사용

3. 마침표가 2번 이상 연속된 부분을 하나의 마침표로 치환 👉🏻find로 '..'가 있는지 검사하고, 있다면 replace함수로 '.'로 대체 + while문 반복

4. 마침표가 처음이나 끝에 위치하면 제거

5. 빈 문자열이라면 a대입

6-1. 16자 이상이라면, 15개의 문자만 남기고 제거

6-2. 6번 시행 후 마침표가 끝에 위치한다면 제거

7. 2자 이하라면 마지막 문자를 길이가 3이 될 때까지 반복

 

나의 풀이

import re

def solution(new_id):
    #1단계
    new_id = new_id.lower()
    
    #2단계
    s = re.compile('([a-z0-9|\-|_|.])')
    new_id = ''.join(s.findall(new_id))
    
    #3단계
    while new_id.find('..') != -1:
        new_id = new_id.replace('..', '.')
    
    #4단계
    find_which = [i for i, value in enumerate(new_id) if value == '.']
    if 0 in find_which:
        new_id = new_id[1:]
    if len(new_id)-1 in find_which:
        new_id = new_id[:len(new_id)]
    
    #5단계
    if new_id == '':
        new_id = 'a'
        
    #6단계-1
    if len(new_id)>=16:
        new_id = new_id[:15]
        
    #6단계-2
    if new_id[-1] == '.':
        new_id = new_id[:len(new_id)-1]
        
    #7단계
    while len(new_id)<=2:
        new_id = ''.join([new_id, new_id[-1]])

    return new_id

주어진 문제를 직관적으로 짜면 됐기 때문에 그다지 어렵진 않았지만, 2단계, 4단계에서 시간을 매우 많이 보내버렸다>_<

 

2단계는 정규표현식을 사용해야 하는데 '다트 게임'(jeonzzang.tistory.com/31?category=954636)문제를 풀었던 기억을 더듬어 작성해봤따 !

여기서 주의할 점은 특수문자 _와 .은 그대로 입력해도 알아서 인식하지만, -의 경우는 앞에 \를 붙여 '이것은 너가 인식해야 할 문자다!'라고 알려줘야 한다.(아마 a-z0-9와 같이 -가 문자가 아닌 하나의 지시문?으로 인식하게 되기 때문인 듯하다)

 

4단계는 위의 코드와 같이, 우선 find_which에 '.'를 포함한 문자열의 위치를 출력해줬다.

이를 통해 0과 len(new_id)가 find_which에 있는지 검사하는 함수를 작성하여 첫 번째 문자열과 마지막 문자열에 '.'를 포함하고 있는지를 파악했다.

주의할 점이 처음에는 new_id[0]과 new_id[-1]이 '.'인지를 확인하는 함수를 작성했는데, 이렇게 되면 new_id가 공백일 때 오류가 발생한다는 것 !

 

다른 풀이들을 통해서 더 간결하고 좋은 코드를 작성할 수 있도록 해보자-!

 

 

다른 풀이

import re

def solution(new_id):
    st = new_id
    st = st.lower()
    st = re.sub('[^a-z0-9\-_.]', '', st)
    st = re.sub('\.+', '.', st)
    st = re.sub('^[.]|[.]$', '', st)
    st = 'a' if len(st) == 0 else st[:15]
    st = re.sub('^[.]|[.]$', '', st)
    st = st if len(st) > 2 else st + "".join([st[-1] for i in range(3-len(st))])
    return st

replace대신 re모듈의 sub함수를 사용했다.

3단계, 4단계, 7단계 코드가 포인트인 것 같다.

 

확실히 아는 만큼 더 멋있는 코드를 작성할 수 있다 ! 자주는 못하더라도 꾸준히 공부하자아 🐣