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

문제 링크 : https://school.programmers.co.kr/learn/courses/30/lessons/72410

프로그래머스

문제 설명은 코드를 작성하기 위한 순서에서 자세히 설명하기 때문에 생략하도록 하겠다.

 

코드를 작성하기 위한 순서는 다음과 같다.

1. 대문자를 소문자로 변환해준다.

2. '.' '_' '-'을 제외한 문자를 제거해준다.

3. 연속되는 '.'을 모두 한 개의 '.'로 변환해준다.

4. 문자열의 시작과 끝에 '.'이 있는 경우, 제거해준다.

5. 문자열의 길이가 16 이상인 경우, 잘라준다.

6. 문자열이 없을 경우 'a'로 채워준다.

7. 길이를 자른 문자열의 시작과 끝에 '.'이 있는 경우 제거해준다.

8. 문자열의 길이가 2이하인 경우, 마지막 문자를 추가해 길이를 늘려준다.

import re #re.sub을 사용하기 위한 import문

def solution(new_id):
    answer = ''
    new_id=new_id.lower() #1번
    moonsa='~!@#$%^&*()=+[{]}:?,<>\'/'
    for i in new_id:
        if i in moonsa:
            new_id=new_id.replace(i,'') #2번
    new_id = re.sub('\.+', '.', new_id) #3번 : '\.+'는 '.'이 여러 개인 것을 의미하고 이를 '.' 하나로 대처한다는 코드이다.
    new_id = new_id.strip('.') #4번
    if len(new_id)>=16: #5번
        new_id=new_id[:15]
    new_id = new_id.strip('.') #7번
    if not new_id: #6번
        new_id='aaa'
    if len(new_id)<=2: #8번
        while len(new_id)<=2:
            new_id+=new_id[-1]
    return new_id

본 문제는 문제의 조건에 따라서 순서대로 차근차근 해결하면 어렵지 않은 문제였다.

다만 연속되는 '.'를 '.' 하나로 변환하는 과정에서 문제를 겪었지만, re.sub이라는 함수를 찾아서 쉽게 해결할 수 있었다.