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

프로그래머스 <신규 아이디>

https://programmers.co.kr/learn/courses/30/lessons/72410?language=python3 

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

---

이중우선순위큐

규칙에 맞지 않은 입력받은 아이디를 규칙에 맞게 바꿔라.

 

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

---

문제 해결

이 문제를 처음에 정규식을 사용하지 않고 풀었을 때는 테스트케이스 2개가 틀렸다.

아무리 생각해도 뭐가 틀린지 모르겠어서 검색해보니 정규식을 사용해야 한다고 했다.

 

파이썬에서 정규표현식을 사용하기 위해 re 패키지를 import해야 한다.

 

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

lower()를 이용하여 모든 대문자를 소문자로 치환한다.

 

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

정규표현식을 이용하여 가능한 문자를 빼고 모두 제거한다.

\w가 알파벳 소문자, 대문자, 숫자를 의미하고 +는 최소 한 번 이상의 반복을 의미한다.

따라서 [\w\_.-]+를 사용하면 알파벳 소문자, 대문자, 숫자, 빼기(-), 밑줄(_), 마침표(.)로만 이루어진 문자열을 의미한다.

이때 이러한 정규표현식에 해당하는 문자열을 구하기 위해 findall()을 사용하면 된다.

 

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

정규표현식을 이용하여 마침표(.)가 2개 이상인 경우를 1개로 치환할 수 있다.

이때 마침표(.)가 2개 이상인 것을 찾기 위해 [.]{2,} 정규표현식을 사용할 수 있다.

sub(a, b, s)함수를 이용하여 s에서 a에 해당하는 문자열을 b로 바꿀 수 있다.

 

나머지 단계는 각각의 단계에 맞게 구현하면 된다.

---

코드

import re
def solution(new_id):
    answer = ''
    new_id = new_id.lower()
    temp = re.findall('[\w\_.-]+', new_id)
    for s in temp:
        answer += s
    answer = re.sub('[.]{2,}', '.', answer)
    size = len(answer)
    if size > 0 and answer[0] == '.':
        answer = answer[1:]
        size -= 1
    if size > 0 and answer[size-1]=='.':
        answer = answer[:size-1]
        size-=1
    if size == 0:
        answer = 'a'
        size = 1
    if size >= 16:
        answer = answer[:15]
        size = 15
        if answer[14] == '.':
            answer = answer[:14]
            size = 14
    elif size <=2:
        for i in range(3-size):
            answer += answer[size-1]
    return answer

---

결과

---