[Programmers] 신규 아이디 추천 - python

프로그래머스의 신규 아이디 추천 문제입니다. 

주어진 문자열을 아래 형식에 맞게 바꾸는 문제입니다.

 

정규식을 활용해 풀었습니다.

 

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

 

def solution(new_id):
	# 1단계
    new_id = new_id.lower()
	
    # 2단계
    new_id = "".join(re.findall("[a-z0-9\.\-\_]", new_id))
    # new_id = re.sub("[^a-z0-9\.\-\_]","",new_id)
    
    # 3단계
    new_id = re.sub("\.+\.", ".", new_id)
	-> .으로 시작해서 .으로 끝나는데 앞 .이 한 개 이상 반복되는 것을 찾아라
    
    # 4단계
    new_id = new_id.strip(".")
	
    # 5단계
    if not new_id:
        new_id = "a"
    
    # 6단계
    if len(new_id) >= 16:
        new_id = new_id[:15]
    # 'a' if len(st) == 0 else st[:15]
    new_id = new_id.strip(".")
	
    # 7단계
    if len(new_id) <= 2:
        new_id = new_id + str(new_id[-1]*(3-len(new_id)))

    answer = new_id
    return answer

 

나는 1단계에서 해당하는 부분들을 찾도록 코드를 작성했다.

 

[a-z0-9\.\-\_]: 소문자, 숫자, ., -, _를 찾아라

 

new_id = "".join(re.findall("[a-z0-9\.\-\_]", new_id))

 

그러나 이렇게 되면 re.findall의 반환 형태가 리스트가 되어 다시 합쳐주어야하는 문제가 발생한다.

 

다른 사람들의 코드를 보니 이 문제를 sub를 사용하여 해결할 수 있었다.

 

[^a-z0-9\.\-\_]: 소문자, 숫자, ., -, _가 아닌 것들을 ""으로 바꿔라

-> new_id = re.sub("[^a-z0-9\.\-\_]","",new_id)

 

 

5단계와 6단계 코드 역시 더 간단히 줄일 수 있었다. 

if not new_id:

        new_id = "a"

   

if len(new_id) >= 16:

        new_id = new_id[:15]

 

-> 'a' if len(st) == 0 else st[:15]

 

re.findall("정규식(찾을 것)", "본 문자열"): 반환 형태 -> 리스트

: 정규식에 해당하는 내용을 본 문자열에서 찾아서 리스트 형태로 반환한다.

 

re.sub("정규식", "바꿀 것", "본 문자열"): 반환 형태 -> 문자열

: 정규식에 해당하는 내용을 본 문자열에서 찾아 바꿀 것으로 바꿔준다.