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

문제

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

문제

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

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

 

처음 작성한 코드

def solution(new_id):
    import re
    new_id = ''.join(re.findall('[a-z0-9-_.]',new_id.lower()))
    while True:
        new_id = new_id.replace('..','.')
        if new_id.find("..") == -1: break
    new_id = list(new_id)
    if new_id[0] == '.': (new_id).pop(0)
    if new_id:
        if new_id[-1] == '.': (new_id).pop()
    if not new_id: new_id.append('a')
    if len(new_id)>=16:
        new_id = new_id[:15]
        if new_id[-1]=='.': new_id.pop()
    if len(new_id)<=2:
        last = new_id[-1]
        while len(new_id)<3:
            new_id.append(last)    
    return ''.join(new_id)

 

코드 리뷰 후

def solution(new_id):
    import re
    new_id = re.sub('[^a-z0-9\-_.]','',new_id.lower())
    new_id = re.sub('\.+','.',new_id)
    new_id = re.sub('^[.]|[.]$','',new_id)
    if not new_id: new_id = 'a'
    if len(new_id)>=16: 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

 

깨달은 점

• re.sub(정규식, 대체할 문자열, 대체될 문자열) 로 쓸 수 있다.
• ^[문자열] -> 문자열로 시작
• [문자열]$ -> 문자열로 끝
• '\d+' -> 0-9까지의 수가 여러번 나옴
• '\.+' -> .이 여러번 나옴