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

신규 아이디 추천 문제

링크

문제 요약

다음과 같이 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이 될 때까지 반복해서 끝에 붙입니다.

나의 풀이

import string, re

def dot(id):
    if (id[0]=='.'): id = id[1:]
    elif(id[-1]=='.'): id = id[:-1]
    return id

def solution(new_id):
    answer = ''
    # 1단계
    id = new_id.lower()
    # 2단계
    symbols = string.punctuation.replace('-', '').replace('_', '').replace('.', '')
    for symbol in symbols: 
    	id = id.replace(symbol, "")
    # 3단계
    id = re.sub('(([.])\\2{1,})', '.', id)
    # 4단계
    id = dot(id)
    # 5단계
    if (len(id) == 0): id="a"
    # 6단계
    if (len(id)>15): id = id[:15]
    id = dot(id)
    # 7단계
    if (len(id)<=2):
        while(len(id) < 3): id = id+id[-1]
    answer=id
    return answer

풀이

링크

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

Key

• 정규식을 쓰면 훨씬 간단하게 풀 수 있는 문제