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

문제 (LV.1)

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

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

 

코드

import re

def dot_deleted(ans):
    if ans[0] == '.':
        ans = ans[1:]
    elif ans[-1] == '.':
        ans = ans[:-1]
    return ans

def solution(new_id):
    answer = ""

    # 1단계 소문자로 변경
    tmp = new_id.lower()

    # 2단계 허용되는 문자만 사용
    for i in tmp:
        if i.isalpha() or i.isdigit() or (i == '-') or (i == '-') or (i == '_') or (i == '.'):
            answer += i

    # 3단계 연속된 '.' 하나로 만들기
    answer = re.sub('[.]+', '.', answer) # 정규식 연속된거 +로 체크

    # 4단계 아이디 처음에 위치한 '.'제거
    answer = dot_deleted(answer)
    # print(answer)
    # if answer[0] == '.':
    #     answer = answer[1:]
    # elif answer[-1] == '.':
    #     answer = answer[:-1]

    # 5단계 빈문자열이면 'a'
    if len(answer) == 0:
        answer = 'a'

    # 6단계 길이 16자 이상이면 처음 15자까지만
    if len(answer) > 15:
        answer = answer[:15]
    answer = dot_deleted(answer)
    
    # 7단계 아이디가 2자 이하라면 마지막문자를 길이가 3이될 때 까지 붙임
    while len(answer) <= 2:
        answer = answer + answer[-1]
    return answer

 

해설

 

단순 문자열 관련 구현이 필요한 문제였다.

특이 사항이라면 4단계인 맨 앞 또는 맨 뒤에 (.) 을 제거하는 과정에서 순서대로 가게된다면, 16번째 자리가 (.)인 문자열은 7가지 조건에 부합하지 않게된다.

따라서 먼저 4단계에서 제거 해준 후에 함수를 구현하여 6단계 이후에 한번 더 (.)을 제거하는 함수를 호출하여 테스트 케이스를 해결하였다.