[프로그래머스] 신규 아이디 추천/ KAKAO BLIND RECRUITMENT / 파이썬

문제

코딩테스트 연습 - 신규 아이디 추천

풀이

1. 파이썬 내장 모듈인 re 모듈을 이용하여 풉니다.

 

코드

import re


def solution(new_id):
    #1단계: 모든 대문자를 대응되는 소문자로 치환
    new_id = new_id.lower()
    
    #2단계: 알파벳 소문자, 숫자, 빼기(-), 밑줄(_), 마침표(.)를 제외한 모든 문자를 제거
    new_id = re.sub('[^\w\.-]', '', new_id)
    
    #3단계: 마침표(.)가 2번 이상 연속된 부분을 하나의 마침표(.)로 치환
    new_id = re.sub('\.{2,}', '.', new_id)
    
    #4단계: 마침표(.)가 처음이나 끝에 위치한다면 제거
    new_id = new_id.strip(".")
    
    #5단계: 빈 문자열이라면, new_id에 "a"를 대입
    new_id = "a" if new_id == "" else new_id
    
    #6단계: 길이가 16자 이상이면, new_id의 첫 15개의 문자를 제외한 나머지 문자들을 모두 제거
    #      만약 제거 후 마침표(.)가 new_id의 끝에 위치한다면 끝에 위치한 마침표(.) 문자를 제
    new_id = new_id[:15].rstrip(".") if len(new_id) > 15 else new_id
    
    #7단계: 길이가 2자 이하라면, new_id의 마지막 문자를 new_id의 길이가 3이 될 때까지 반복해서 끝에 붙임
    while len(new_id) < 3:
        new_id += new_id[-1]
        
    return new_id