프로그래머스 72410번 - 신규 아이디 추천 (★★ / O / 1) : Python

개요

• 풀이 시간 : 20분
• 시간 제한 : -
• 메모리 제한 : -
• 기출 : 2021 KAKAO BLIND RECRUITMENT
• 링크 : https://programmers.co.kr/learn/courses/30/lessons/72410
  
  

문제

카카오에 입사한 신입 개발자 네오는 "카카오계정개발팀"에 배치되어, 카카오 서비스에 가입하는 유저들의 아이디를 생성하는 업무를 담당하게 되었습니다. "네오"에게 주어진 첫 업무는 새로 가입하는 유저들이 카카오 아이디 규칙에 맞지 않는 아이디를 입력했을 때, 입력된 아이디와 유사하면서 규칙에 맞는 아이디를 추천해주는 프로그램을 개발하는 것입니다.

다음은 카카오 아이디의 규칙입니다.

• 아이디의 길이는 3자 이상 15자 이하여야 합니다.
• 아이디는 알파벳 소문자, 숫자, 빼기(-), 밑줄(_), 마침표(.) 문자만 사용할 수 있습니다.
• 단, 마침표(.)는 처음과 끝에 사용할 수 없으며 또한 연속으로 사용할 수 없습니다.

"네오"는 다음과 같이 7단계의 순차적인 처리 과정을 통해 신규 유저가 입력한 아이디가 카카오 아이디 규칙에 맞는 지 검사하고 규칙에 맞지 않은 경우 규칙에 맞는 새로운 아이디를 추천해 주려고 합니다.

신규 유저가 입력한 아이디가 new_id 라고 한다면,

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

신규 유저가 입력한 아이디를 나타내는 new_id가 매개변수로 주어질 때, "네오"가 설계한 7단계의 처리 과정을 거친 후의 추천 아이디를 return 하도록 solution 함수를 완성해 주세요.

제한 사항

new_id는 길이 1 이상 1,000 이하인 문자열입니다.
new_id는 알파벳 대문자, 알파벳 소문자, 숫자, 특수문자로 구성되어 있습니다.
new_id에 나타날 수 있는 특수문자는 -_.~!@#$%^&*()=+[{]}:?,<>/ 로 한정됩니다.

해결방법

문제 이해하기

뭐.. 그냥 구현 형태로도 문제를 풀 수 있겠지만 그건 너무 불필요한 방법이고.. 이 문제는 정규표현식을 이용하는 것이 좋다. 이 문제도 실제 시험에서 풀었는데 그때는 정규표현식을 사용할 줄 잘 몰라서 일일히 구현했던 경험이 있다. 첫번째 문제였던거 같은데... 가만 보면 카카오는 문자열과 정규표현식을 좋아하나보다.

re.sub

여기서는 re 라이브러리 중에 sub 함수를 사용하면 된다. sub(정규표현식, 대체할 문자(열), 전체 문자열) 이런식으로 사용하면 된다. 그러면 해당 정규표현식에 맞는 녀석을 찾아서 대체할 문자(열)로 교체해준다.

Python

내 코드

import re

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

추천 코드

import re

def solution(new_id):
    st = new_id
    st = st.lower()
    st = re.sub('[^a-z0-9\-_.]', '', st)
    st = re.sub('\.+', '.', st)
    st = re.sub('^[.]|[.]$', '', st)
    st = 'a' if len(st) == 0 else st[:15]
    st = re.sub('^[.]|[.]$', '', st)
    st = st if len(st) > 2 else st + "".join([st[-1] for i in range(3-len(st))])
    return st

# 참고 : https://programmers.co.kr/learn/courses/30/lessons/72410/solution_groups?language=python3&type=all

내 코드랑 얼추 비슷하다. 다만 파이썬 한줄 쓰기 기능을 잘 이용했다고 볼 수 있겠네.

✍️ 이처럼 정규표현식은 카카오나 기타 넥X, 라X 등 다른 기업 코딩테스트에서도 많이 출제되므로 반드시 잘 알아둘 필요가 있다.