[프로그래머스] 신규 아이디 추천 - 파이썬/python

문제

https://school.programmers.co.kr/learn/courses/30/lessons/72410

프로그래머스

주어진 규칙(7단계)에 맞는 아이디를 추천해주는 문제다.

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

 

예시

Example 1:

new_id: "...!@BaT#*..y.abcdefghijklm"
result: "bat.y.abcdefghi"

 

Example 2:

new_id: "z-+.^."
result: "z--"

 

Example 3:

new_id: "=.="
result: "aaa"

Example 4:

new_id: "123_.def"
result: "123_.def"

 

Example 5:

new_id: "abcdefghijklmn.p"
result: "abcdefghijklmn"

 

풀이

각 단계별로 차근차근 풀어주고 2, 3 단계는 정규 표현식을 사용해서 풀었다.

import re

def solution(new_id):
    # 1단계 소문자 치환
    answer = new_id.lower()
    # 2단계 문자, 숫자, -, _, . 제외 모두 제거
    answer = re.sub('[^a-z0-9\-_.]', '', answer)
    # 3단계 ... -> . 치환
    answer = re.sub('\.{2,}', '.', answer)
    # 4단계 . 처음이나 끝 제거
    answer = answer.strip('.')
    # 5단계 빈 문자열이면 -> a
    if len(answer) == 0:
        answer = 'a'
    # 6단계 길이가 16 이상이면 15개까지만, 끝에 . 제거
    answer = answer[:15].strip('.')
    # 7단계 길이가 2 이하라면 마지막 문자를 3이 될때까지 반복
    while len(answer) < 3:
        answer += answer[-1]
    
    return answer

 

👉 정규표현식 re.sub(pattern, replace, text) 활용하기
정규식을 이용해 문자열을 치환하는 방법으로 text 중 pattern에 해당하는 부분을 replace로 대체한다.

 

특수문자	설명
.	한 개의 임의의 문자를 나타냅니다.
?	앞의 문자가 존재할 수도 있고, 존재하지 않을 수도 있습니다. (문자가 0개 또는 1개)
*	앞의 문자가 무한개로 존재할 수도 있고, 존재하지 않을 수도 있습니다. (문자가 0개 이상)
+	앞의 문자가 최소 한 개 이상 존재합니다. (문자가 1개 이상)
^	뒤의 문자열로 문자열이 시작됩니다.
$	앞의 문자열로 문자열이 끝납니다.
{숫자1, 숫자2}	숫자1 이상 숫자2 이하만큼 반복합니다. ?, *, +를 이것으로 대체할 수 있습니다.
[ ]	대괄호 안의 문자들 중 한 개의 문자와 매치합니다. [amk]라고 한다면 a 또는 m 또는 k 중 하나라도 존재하면 매치를 의미합니다. [a-z]와 같이 범위를 지정할 수도 있습니다. [a-zA-Z]는 알파벳 전체를 의미하는 범위이며, 문자열에 알파벳이 존재하면 매치를 의미합니다.
[^문자]	해당 문자를 제외한 문자를 매치합니다.

 

다른 사람의 풀이(정규식)

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

• 2, 3 단계 뿐만 아니라 나머지도 정규 표현식을 사용해서 풀어줬다.

 

 

참고 자료

https://wikidocs.net/21703