[Python] 신규 아이디 추천 - 2021 KAKAO BLIND RECRUITMENT

<문제 링크>

 

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

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

 

---

<풀이>

 

이 문제는 정규화를 사용해 문자열을 변환해주는 re 패키지를 사용하면 쉽게 풀 수 있다.

문제에서 정의하는 신규 아이디 추천 알고리즘은 다음과 같다.

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이 될 때까지 반복해서 끝에 붙입니다.

그래서 위에 조건대로 하나씩 수행해주면 된다.

import re

def solution(new_id):
    # 1
    new_id = new_id.lower()
    # 2
    new_id = re.sub("[^a-z0-9-_.]", "", new_id)
    # 3
    new_id = re.sub("[.]+", ".", new_id)
    # 4
    new_id = re.sub("^\.", "", new_id)
    new_id = re.sub("\.$", "", new_id)
    # 5
    if len(new_id) == 0:
        new_id = 'a'
    # 6
    new_id = new_id[:15]
    new_id = re.sub("\.$", "", new_id)
    # 7
    while len(new_id) < 3:
        new_id += new_id[len(new_id) - 1]
    
    return new_id