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

2021 KAKAO BLIND RECRUITMENT

 

Lv.1 문자열 유형

 

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):
    new_id = new_id.lower() # step 1
    new_id = re.sub(r'[^0-9a-z\-_\.]', '', new_id) # step 2
    new_id = list(filter(None, new_id.split('.'))) # step 3 
    new_id = new_id.strip('.') # step 4
    new_id = 'a' if not new_id else new_id # step 5
    new_id = new_id[:15].rstrip('.') if len(new_id) >= 15 else new_id # step 6
    new_id = new_id + new_id[-1] * (3 - len(new_id)) if len(new_id) <= 2 else new_id # step 7
    
    return new_id

 

시간 복잡도와 공간복잡도는 고려하지 않고 최대한 한 라인으로 단계를 클리어하였다.

시간 복잡도를 고려한다면, step 3 부터 for 문으로 풀어야 할 것으로 생각된다.

다만, 아래 공식 풀이와 같이 입력 문자열 길이 제한이 1000이므로 제한시간 초과는 나오지 않을 것 같다.

 

 

Kakao Tech 블로그 공식 풀이

 

1번 문제는 가장 낮은 난이도에 해당하는 일명 몸풀기 문제입니다. 1단계~7단계에서 지시하는 그대로 구현하면 되기 때문에, 특별한 알고리즘보다는 정확한 구현이 필요한 문제입니다.

본 문제는 new_id의 길이가 1,000으로 매우 작습니다. 따라서, new_id의 길이를 n이라고 할 때, O(n^2) 성능의 알고리즘으로 구현해도 제한 시간 내 답을 구할 수 있습니다.

보다 효율적으로 구현한다면 O(n) 성능의 알고리즘으로 구현을 할 수도 있습니다. new_id에서 필요 없는 문자들을 직접 제거하지 말고, new_id를 앞에서부터 검사하면서 유효한 문자(제거되지 않아야 할 문자)만 추려서 새로운 문자열 변수(new_id_1)에 붙여나가는 방법을 사용하면 됩니다.

 

 

출처: https://tech.kakao.com/2021/01/25/2021-kakao-recruitment-round-1/