[프로그래머스] 신규 아이디 추천 (python)

https://school.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 값이 "...!@BaT#*..y.abcdefghijklm" 라면, 위 7단계를 거치고 나면 new_id는 아래와 같이 변경됩니다.

1단계 대문자 'B'와 'T'가 소문자 'b'와 't'로 바뀌었습니다.
"...!@BaT#*..y.abcdefghijklm" → "...!@bat#*..y.abcdefghijklm"

2단계 '!', '@', '#', '*' 문자가 제거되었습니다.
"...!@bat#*..y.abcdefghijklm" → "...bat..y.abcdefghijklm"

3단계 '...'와 '..' 가 '.'로 바뀌었습니다.
"...bat..y.abcdefghijklm" → ".bat.y.abcdefghijklm"

4단계 아이디의 처음에 위치한 '.'가 제거되었습니다.
".bat.y.abcdefghijklm" → "bat.y.abcdefghijklm"

5단계 아이디가 빈 문자열이 아니므로 변화가 없습니다.
"bat.y.abcdefghijklm" → "bat.y.abcdefghijklm"

6단계 아이디의 길이가 16자 이상이므로, 처음 15자를 제외한 나머지 문자들이 제거되었습니다.
"bat.y.abcdefghijklm" → "bat.y.abcdefghi"

7단계 아이디의 길이가 2자 이하가 아니므로 변화가 없습니다.
"bat.y.abcdefghi" → "bat.y.abcdefghi"

따라서 신규 유저가 입력한 new_id가 "...!@BaT#*..y.abcdefghijklm"일 때, 네오의 프로그램이 추천하는 새로운 아이디는 "bat.y.abcdefghi" 입니다.

---

[문제]

신규 유저가 입력한 아이디를 나타내는 new_id가 매개변수로 주어질 때, "네오"가 설계한 7단계의 처리 과정을 거친 후의 추천 아이디를 return 하도록 solution 함수를 완성해 주세요.

[제한사항]

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

---

[입출력 예]

no	new_id	result
예1	"...!@BaT#*..y.abcdefghijklm"	"bat.y.abcdefghi"
예2	"z-+.^."	"z--"
예3	"=.="	"aaa"
예4	"123_.def"	"123_.def"
예5	"abcdefghijklmn.p"	"abcdefghijklmn"

 

---

✔️ 입출력 예 #1
문제의 예시와 같습니다.

 

✔️ 입출력 예 #2
7단계를 거치는 동안 new_id가 변화하는 과정은 아래와 같습니다.

1단계 변화 없습니다.
2단계 "z-+.^." → "z-.."
3단계 "z-.." → "z-."
4단계 "z-." → "z-"
5단계 변화 없습니다.
6단계 변화 없습니다.
7단계 "z-" → "z--"

 

✔️ 입출력 예 #3
7단계를 거치는 동안 new_id가 변화하는 과정은 아래와 같습니다.

1단계 변화 없습니다.
2단계 "=.=" → "."
3단계 변화 없습니다.
4단계 "." → "" (new_id가 빈 문자열이 되었습니다.)
5단계 "" → "a"
6단계 변화 없습니다.
7단계 "a" → "aaa"

 

✔️ 입출력 예 #4
1단계에서 7단계까지 거치는 동안 new_id("123_.def")는 변하지 않습니다. 즉, new_id가 처음부터 카카오의 아이디 규칙에 맞습니다.

 

✔️ 입출력 예 #5
1단계 변화 없습니다.
2단계 변화 없습니다.
3단계 변화 없습니다.
4단계 변화 없습니다.
5단계 변화 없습니다.
6단계 "abcdefghijklmn.p" → "abcdefghijklmn." → "abcdefghijklmn"
7단계 변화 없습니다.

---

🗒 코드

import re


def solution(new_id):
    # 1. convert uppercase to lowercase
    answer = new_id.lower()
    # 2. remove characters except lowercase, num, -, _, .
    answer = re.sub('[^a-z0-9\-\_\.]', '', answer)
    # 3. if new_id contains continuous periods, replace only one period
    answer = re.sub(r'\.{2,}', '.', answer)
    # 4. remove periods at both ends
    answer = answer.strip('.')
    # 5. if answer is empty string, then substitute answer 'a'
    if len(answer) == 0:
        answer = 'a'
    # 6. if length of answer is 16 or more, truncate to answer 15
    if len(answer) > 15:
        answer = answer[:15].rstrip('.')
    # 7. if length of answer is 2 or less, concatenate the last character of answer until the length is 3
    while len(answer) < 3:
        answer += answer[-1]
    return answer

각 단계별로 차근차근 코드를 짰다.

단계 1에서는 대문자를 모두 소문자로 바꿔준다. 파이썬의 lower() 함수를 사용하면 간단하고 쉽게 변환 가능하다.

단계 2에서는 소문자, 숫자, -, _, .를 제외하고 그 외의 문자들은 제거한다. 정규식을 사용하여 제거하였다.

단계 3에서는 연속된 마침표에 대해 처리를 했다. 단계 2와 마찬가지로 정규식을 사용했다.

단계 4에서는 처음과 끝에 있는 마침표를 제거했다. strip() 함수를 이용하면 되는데, 괄호 안에 아무것도 지정해주지 않으면 공백(띄어쓰기)를 제거해주고, 괄호 안에 문자를 지정하게 되면 해당 문자를 제거해준다. 마침표를 지우면 되는 것이므로, 괄호 안에 마침표를 넣어주었다.

단계 5에서는 단계 1부터 단계 4까지의 처리를 거친 문자열이 빈 문자열이 된 경우를 처리한다. 만약 빈 문자열인 경우 'a'로 대치한다.

단계 6에서는 문자열의 길이가 16 이상인 것을 처리한다. 만약 문자열의 길이가 16 이상이면, 문자열의 길이가 15인 것 까지 자른다. 이 때 문자열의 슬라이싱을 사용했고, 오른쪽 문자열의 마침표를 rstrip()로 처리해준다.

단계 7에서는 문자열의 길이가 3보다 적으면, 해당 문자열의 끝 문자를 반복해서 붙여준다. 단계 6과 마찬가지로 슬라이싱을 사용하여 처리해주었다.

---

🤔 후기

보자마자 정규식을 사용하면 되겠구나 했지만, 정규식이 아주 능숙한 편이 아니라 꽤 많은 시간을 잡아먹었다.

정규식은 왜 쓰고 나면 간단한데, 정규식을 쓰기까지가 힘이 드는지...어려운 구문도 아니었어서 자괴감이...하다보면 늘겠지 뭐

---

통과