[프로그래머스] 72410. 신규 아이디 추천

문제 링크

[프로그래머스] 72410. 신규 아이디 추천

풀이 과정

정규식 패턴을 찾기 위해 re 모듈을 사용했습니다.

4단계 new_id에서 마침표(.)가 처음이나 끝에 위치한다면 제거합니다.

정규식 패턴 /^\.|\.$/ 을 사용해 맨 처음과 맨 마지막에 마침표가 나올 경우 한 번에 제거했습니다.

5단계 new_id가 빈 문자열이라면, new_id에 “a”를 대입합니다.
6단계 new_id의 길이가 16자 이상이면, new_id의 첫 15개의 문자를 제외한 나머지 문자들을 모두 제거합니다.
만약 제거 후 마침표(.)가 new_id의 끝에 위치한다면 끝에 위치한 마침표(.) 문자를 제거합니다.

new_id = "a" if len(new_id) == 0 else re.sub("\.$", "", new_id[:15])

5단계와 6단계를 위와 같이 한 번에 진행했습니다.

• 빈 문자열이면 a를 대입한다.
• 문자열 길이가 16 이상이라면, 앞에서 15개 문자를 자른 뒤 끝에 마침표가 존재하면 지운다.

코드

import re

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

    return new_id

print(solution("...!@BaT#*..y.abcdefghijklm"))
print(solution("z-+.^."))
print(solution("=.="))
print(solution("123_.def"))
print(solution("abcdefghijklmn.p"))