Programmers / 신규 아이디 추천 / Python 파이썬

 

*문제 출처는 프로그래머스에 있습니다.

문제 제목: 신규 아이디 추천

문제 사이트: https://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에 나타날 수 있는 특수문자는 -_.~!@#$%^&*()=+[{]}:?,<>/ 로 한정됩니다.

 

---

나의 풀이

def solution(new_id):
    new_id = new_id.lower()  # 1단계
    string = ""
    for i in new_id:  # 2단계
        if i.isnumeric() or i.islower():
            string += i
        elif i == '-':
            string += i
        elif i == '_':
            string += i
        elif i == '.':
            string += i

    while '..' in string:  # 3단계
        string = string.replace('..', '.')

    # 4단계
    if string[0] == ".":
        if len(string) >= 2:
            string = string[1:]
        else:
            string = "."
            
    if string[-1] == ".":
        string = string[:-1]


    if string == "":  # 5단계
        string += "a"

    if len(string) >= 16:  # 6단계
        string = string[:15]
        if string[-1] == ".":
            string = string[:14]
            
    while len(string) < 3:
        string += string[-1]

    return string

 

 

모범답안

 

re 정규식 사용 

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

re.sub(pattern, repl, string, count=0, flags=0) 

 

string에서 겹치지 않는 pattern의 가장 왼쪽 일치를 repl로 치환하여 얻은 문자열을 반환

 

 re.sub('[^A-Za-z0-9]', '', ‘(Hello123)’) 


→ Hello123 (문자, 숫자가 아닌 부분을 ''로 대체)

출처: https://kibua20.tistory.com/199 [모바일 SW 개발자가 운영하는 블로그]

---

※ 알아야 할 것

- 무리하게 remove, pop을 사용해서 여러 길로 돌아가는 것 보다는 원초적으로 string 문자열을 이용해 문제를 풀면 쉽게 풀 수 있다.