문제 설명
카카오에 입사한 신입 개발자 네오는 "카카오계정개발팀"에 배치되어, 카카오 서비스에 가입하는 유저들의 아이디를 생성하는 업무를 담당하게 되었습니다. "네오"에게 주어진 첫 업무는 새로 가입하는 유저들이 카카오 아이디 규칙에 맞지 않는 아이디를 입력했을 때, 입력된 아이디와 유사하면서 규칙에 맞는 아이디를 추천해주는 프로그램을 개발하는 것입니다.
다음은 카카오 아이디의 규칙입니다.
- 아이디의 길이는 3자 이상 15자 이하여야 합니다.
- 아이디는 알파벳 소문자, 숫자, 빼기(-), 밑줄(_), 마침표(.) 문자만 사용할 수 있습니다.
- 단, 마침표(.)는 처음과 끝에 사용할 수 없으며 또한 연속으로 사용할 수 없습니다.
"네오"는 다음과 같이 7단계의 순차적인 처리 과정을 통해 신규 유저가 입력한 아이디가 카카오 아이디 규칙에 맞는 지 검사하고 규칙에 맞지 않은 경우 규칙에 맞는 새로운 아이디를 추천해 주려고 합니다.
신규 유저가 입력한 아이디가 new_id 라고 한다면,
1단계 newid의 모든 대문자를 대응되는 소문자로 치환합니다.
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 함수를 완성해 주세요.
제한사항
newid는 길이 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" |
내 풀이
import re
badchars = '~!@#$%^&*()=+[{]}:?,<>/'
def solution(new_id):
answer = new_id.lower()
answer = ''.join([x for x in answer if x not in badchars])
answer = re.sub(r'\.{2,}', '.', answer)
if answer and answer[0] == '.':
answer = answer[1:]
if answer and answer[-1] == '.':
answer = answer[:-1]
if len(answer)==0:
answer = 'a'
elif len(answer)>=16:
answer = answer[:15]
if answer[-1] == '.':
answer = answer[:-1]
if len(answer)<=2:
while len(answer) < 3:
answer += answer[-1]
#answer = answer[0:3]
return answer코멘트
우선 중요한 점은 문제를 끝까지 다 읽자는 것이다. 아래의 입출력 예가 잘 돌아가도록만 코드를 작성해도 대부분의 테스트 케이스는 통과할 수 있다.
그리고 이렇게 문자열을 다루는 문제에서는 re 라이브러리 사용법을 잘 알아두도록 하자. 마침표가 2개 이상인 부분을 어떻게 찾아낼 수 있는지 몰랐는데, re를 이용하니 저렇게 간단히 찾아낼 수 있었다.
이 문제에서 내가 겪은 가장 큰 문제는 입출력 예시는 다 옳은 답을 내어놔도 2, 22, 23번 테스트 케이스에서 런타임 에러가 떴다는 것인데, 이는 빈 문자열에서 인덱스를 찾으려고 하면 나타나는 에러였다. 3단계까지 거치고 4단계를 실행할 때 빈 문자열이 되었다면 answer[0] 혹은 answer[1]을 할 때 IndexError가 나타나면서 런타임 에러가 뜬다. 따라서 먼저 answer이 빈 문자열인지를 확인해주는 if answer를 추가해 주었다.
그리고 7단계 마지막 문자를 추가해주는 부분에서, 문자열의 마지막 문자만 1개씩 계속 추가되는 구조이므로 while len(answer) < 3:을 작성했다면 결과 answer은 무조건 3글자가 될 것이므로 굳이 인덱싱해서 answer = answer[0:3]을 할 필요가 없다.
또, if와 elif의 사용에도 주의하자. elif는 완전히 배타적인 조건에서 사용해야 하고, 만약 겹치는 조건이 생긴다면 if, if로 따로 작성하는 것이 좋다. 처음에 if answer[0]=='.':, elif answer[-1]=='.'로 작성했다가 .abcde.와 같은 상황에 대처하지 못하는 문제가 있었다.
References
https://school.programmers.co.kr/learn/challenges?order=recent
