[프로그래머스] 신규 아이디 추천 / Python / 2021 KAKAO BLIND RECRUITMENT

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

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

---

---

Solution

import re

op = ['-', '_', '.']

def solution(new_id):
    new_id = new_id.lower()
    tmp = ''
    for c in new_id:
        if c.isalpha() or c.isdigit() or c in op:
            tmp += c
    new_id = tmp
    new_id = re.sub('\.\.+', '.', new_id)
    new_id = re.sub('^\.|\.$', '', new_id)
    if not new_id:
        new_id = "a"
    if len(new_id) > 15:
        new_id = new_id[:15]
        new_id = re.sub('\.$', '', new_id)
    if len(new_id) <= 2:
        last = new_id[-1]
        while len(new_id) != 3:
            new_id += last
    return new_id

 

문제에서 주어진 단계를 그대로 구현만 하면 되는 문제였다.

정규식을 좀 더 자유롭게 다룰 수 있으면 시간 단축이 많이 될 것 같다.

 

1단계 - new_id의 모든 대문자를 대응되는 소문자로 치환합니다.

 

new_id = new_id.lower()

 

2단계 - new_id에서 알파벳 소문자, 숫자, 빼기(-), 밑줄(_), 마침표(.)를 제외한 모든 문자를 제거합니다.

 

op = ['-', '_', '.']

...

tmp = ''
for c in new_id:
	if c.isalpha() or c.isdigit() or c in op:
		tmp += c
new_id = tmp

 

new_id의 문자를 하나씩 검사하고 문자가 알파벳이거나 숫자거나 op에 들어있는 문자일 경우 tmp에 더해준다.

 

3단계 - new_id에서 마침표(.)가 2번 이상 연속된 부분을 하나의 마침표(.)로 치환합니다.

 

import re

new_id = re.sub('\.\.+', '.', new_id)

 

여기서 정규식을 썼다. sub을 통해 마침표가 2개 이상 연속될 경우 하나의 마침표로 치환하게끔 했다. \.은 마침표 자체를 나타내며, *은 정규표현식 메타문자로서 반복을 나타낼 때 사용한다. 이전 문자\.가 1번 이상 반복될 경우를 나타낸다.

 

4단계 - new_id에서 마침표(.)가 처음이나 끝에 위치한다면 제거합니다.

 

new_id = re.sub('^\.|\.$', '', new_id)

 

똑같이 sub으로 치환을 했고, ^메타문자는 문자열의 맨 처음과 일치함을 의미한다. |는 or과 똑같은 의미이다. $는 문자열의 맨 끝과 일치함을 의미한다. 따라서 문자열의 맨 처음에 마침표가 있거나, 맨 끝에 마침표가 있으면 제거한다는 의미이다.

 

5단계 - new_id가 빈 문자열이라면, new_id에 "a"를 대입합니다.

 

if not new_id:
    new_id = "a"

 

6단계 - new_id의 길이가 16자 이상이면, new_id의 첫 15개의 문자를 제외한 나머지 문자들을 모두 제거합니다. 만약 제거 후 마침표(.)가 new_id의 끝에 위치한다면 끝에 위치한 마침표(.) 문자를 제거합니다.

 

if len(new_id) > 15:
    new_id = new_id[:15]
    new_id = re.sub('\.$', '', new_id)

 

7단계 - new_id의 길이가 2자 이하라면, new_id의 마지막 문자를 new_id의 길이가 3이 될 때까지 반복해서 끝에 붙입니다.

 

if len(new_id) <= 2:
    last = new_id[-1]
    while len(new_id) != 3:
        new_id += last