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

 

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

 

✅ 접근

지문 길이가 길어 당황했지만, 적힌 내용을 그대로 코드로 옮기면 되는 어렵지 않은 문제입니다.

문자열의 앞단과 뒷단에서 빼는 작업이 필요하여 deque() 을 사용했습니다.

 

💻 코드

 

from collections import deque


def solution(new_id):
    # 1단계
    new_id = new_id.lower()

    # 2단계
    second = ''
    for l in new_id:
        if l.isdigit() or l.isalpha() or l in ['-', '_', '.']:
            second += l

    # 3단계
    while True:
        if '..' in second:
            second = second.replace('..', '.')
        else:
            break

    # 4단계
    second = deque(second)
    if second[0] == '.':
        second.popleft()

    if second and second[-1] == '.':
        second.pop()

    # 5단계
    if not second:
        second = 'a'

    # 6단계
    second = list(second)
    if len(second) >= 16:
        second = second[0:15]
        if second[-1] == '.':
            second = second[0:14]

    # 7단계 
    while len(second) <= 2:
        second += second[-1]

    return ''.join(second)

 

❄️ 주의

저는 테스트케이스 3,4,5,11,15에서 실패했는데, 이유는 4단계에 있었습니다.

(a)첫번째 문자 혹은 (b)마지막 문자가 "."인 경우에 popleft(), pop() 을 해주어야 하는데,

if ~ else if  구문을 썼기 때문에 (a), (b) 모든 경우가 아닌 (a) 에 대해서만 실행됐습니다.

if~ else if 구문이 아닌 2개의 if 구문으로 나누어 문제를 해결할 수 있었습니다.

 

또한 첫번째 if 문이 실행된 후, 문자열이 존재하지 않게 되면 두번째 if 문에서 곧바로 pop() 할 문자가 없어집니다. 

때문에 second 라는 문자열이 먼저 존재하는지(if second and ...) 확인 후 조건문을 실행하도록 수정해야 합니다.

 

 

🦁 정규 표현식 풀이

정규표현식을 이용한 풀이도 존재합니다.

정규 표현식을 사용하면 .isdigit() 같은 메서드를 모르더라도 문제를 쉽게 풀 수 있습니다.

def solution2(new_id):

    # 1 : 소문자 치환
    new_id = new_id.lower()

    # 2 : 특정 문자 외 제거
    answer = re.sub('[^a-z\d\-\_\.]', '', new_id)

    # 3 : 마침표 2번 연속은 하나로 치환
    answer = re.sub('\.\.+', '.', answer)

    # 4 : 양 끝 마침표 제거
    answer = re.sub('^\.|\.$', '', answer)

    # 5 : 빈 문자열일 경우 'a' 추가
    if not answer:
        answer = 'a'

    # 6 : 길이가 16자 이상이면 1~15자만 남기기 & 맨 끝 마침표 제거
    answer = re.sub('\.$', '', answer[0:15])

    # 7 : 길이가 2 이하면, 3이 될 때까지 해당 문자열의 마지막 글자를 반복해서 붙인다.
    while len(answer) < 3:
        answer += answer[-1]

    return answer