Python/Coding Test

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

semperparatus 2024. 2. 26. 09:33

https://school.programmers.co.kr/learn/courses/30/lessons/72410

 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr


코드

def solution(new_id):
    answer = new_id
    
    # 1단계 : 대문자 -> 소문자 치환
    answer = answer.lower()
            
    # 2단계 : 문자 제거
    remove_list = list('~!@#$%^&*()=+[{]}:?,<>/')
    for j in remove_list:
        answer = answer.replace(j, '')
    
    # 3단계 : . 치환
    while '..' in answer:
        answer = answer.replace('..', '.')
    
    # 4단계 : 마침표 처음, 끝 제거
    if len(answer) >= 1 and answer[0] == '.':
        answer = answer[1:]
    if len(answer) >= 1 and answer[-1] == '.':
        answer = answer[:-1]
        
    # 5단계 : 빈 문자열 대입
    if not answer:
        answer = 'a'
        
    # 6단계 : 문자열 길이가 길다면
    if len(answer) >= 16:
        answer = answer[:15]
        if answer[-1] == '.':
            answer = answer[:14]
    
    # 7단계 : 문자열 길이가 짧다면
    while len(answer) < 3:
        answer += answer[-1]        
            
    return answer

풀이

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

 

신규 유저가 입력한 new_id에 대해 다음과 같은 처리를 해주어야 한다.

1단계는 대문자 -> 소문자 치환이기 때문에 문자열.lower() 함수를 사용하면 된다. lower(), upper(), islower(), isupper() 등 대소문자 관련 함수를 잘 기억해서 활용하도록 하자.

2단계는 주어진 문자를 제외한 모든 문자를 제거하는 문제이다. 우선 제한사항에서 나타날 수 있는 특수문자가 -_.~!@#$%^&*()=+[{]}:?,<>/ 라고 하였다. 남겨야 하는 특수 문자인 '-', '_', '.'를 제외한 특수문자 리스트를 만들고, 해당 리스트를 for문을 돌면서 빈 문자열로 치환해주면 된다.

for c in new_id:
	if c.isalpha() or c.isdigit() or c in ['-', '_', '.']:
    	answer += c

2단계 다른 문제 풀이를 찾아보니 나와 정반대의 방식으로 풀이를 한 것도 있었다. isalpha(), isdigit() 등 해당 문자열이 알파벳, 숫자인지 확인하는 함수와 in 연산자를 활용해 구현할 수 있다는 것도 알게 되었다.

3단계는 마침표 2번 이상 연속된 부분을 하나의 마침표로 치환하는 문제인데, 정규표현식을 사용하고 싶었지만 문법이 기억나지 않았다. 다른 방법을 고민하다가 while 반복문 + replace() 함수를 동시에 사용해 반복된 마침표를 하나의 마침표로 치환할 수 있었다.

4단계는 처음과 끝의 마침표를 제거할 때 슬라이싱을 사용했는데 인덱싱 범위를 벗어나는 것을 주의해야 한다. 따라서 해당 문자열의 길이가 1보다 크거나 같은지 확인 후 슬라이싱을 해주어야 한다.

5, 6, 7단계는 위 단계만 잘 따라온다면 큰 무리없이 따라올 수 있었다. 인덱싱 및 슬라이싱 할 때 인덱스 번호만 주의해주면 된다.