[Python] Lv1. 신규 아이디 추천

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

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

 

 

 

 

 

 

문제 설명

카카오에 입사한 신입 개발자 네오는 "카카오계정개발팀"에 배치되어, 카카오 서비스에 가입하는 유저들의 아이디를 생성하는 업무를 담당하게 되었습니다. "네오"에게 주어진 첫 업무는 새로 가입하는 유저들이 카카오 아이디 규칙에 맞지 않는 아이디를 입력했을 때, 입력된 아이디와 유사하면서 규칙에 맞는 아이디를 추천해주는 프로그램을 개발하는 것입니다.

다음은 카카오 아이디의 규칙입니다.

• 아이디의 길이는 3자 이상 15자 이하여야 합니다.
• 아이디는 알파벳 소문자, 숫자, 빼기(-), 밑줄(_), 마침표(.) 문자만 사용할 수 있습니다.
• 단, 마침표(.)는 처음과 끝에 사용할 수 없으며 또한 연속으로 사용할 수 없습니다.

"네오"는 다음과 같이 7단계의 순차적인 처리 과정을 통해 신규 유저가 입력한 아이디가 카카오 아이디 규칙에 맞는지 검사하고 규칙에 맞지 않은 경우 규칙에 맞는 새로운 아이디를 추천해 주려고 합니다.
신규 유저가 입력한 아이디가 new_id 라고 한다면,

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

1단계, 2단계, 3단계 독자 가능하다.

예를 들어, new_id 값이 "...!@BaT#*..y.abcdefghijklm" 라면, 위 7단계를 거치고 나면 new_id는 아래와 같이 변경됩니다.

1단계 대문자 'B'와 'T'가 소문자 'b'와 't'로 바뀌었습니다.
"...!@BaT#*..y.abcdefghijklm" → "...!@bat#*..y.abcdefghijklm"

2단계 '!', '@', '#', '*' 문자가 제거되었습니다.
"...!@bat#*..y.abcdefghijklm" → "...bat..y.abcdefghijklm"

3단계 '...'와 '..' 가 '.'로 바뀌었습니다.
"...bat..y.abcdefghijklm" → ".bat.y.abcdefghijklm"

4단계 아이디의 처음에 위치한 '.'가 제거되었습니다.
".bat.y.abcdefghijklm" → "bat.y.abcdefghijklm"

5단계 아이디가 빈 문자열이 아니므로 변화가 없습니다.
"bat.y.abcdefghijklm" → "bat.y.abcdefghijklm"

6단계 아이디의 길이가 16자 이상이므로, 처음 15자를 제외한 나머지 문자들이 제거되었습니다.
"bat.y.abcdefghijklm" → "bat.y.abcdefghi"

7단계 아이디의 길이가 2자 이하가 아니므로 변화가 없습니다.
"bat.y.abcdefghi" → "bat.y.abcdefghi"

따라서 신규 유저가 입력한 new_id가 "...!@BaT#*..y.abcdefghijklm"일 때, 네오의 프로그램이 추천하는 새로운 아이디는 "bat.y.abcdefghi" 입니다.

 

[문제]

신규 유저가 입력한 아이디를 나타내는 new_id가 매개변수로 주어질 때, "네오"가 설계한 7단계의 처리 과정을 거친 후의 추천 아이디를 return 하도록 solution 함수를 완성해 주세요.

 

제한 사항

new_id는 길이 1 이상 1,000 이하인 문자열입니다.
new_id는 알파벳 대문자, 알파벳 소문자, 숫자, 특수문자로 구성되어 있습니다.
new_id에 나타날 수 있는 특수문자는 -_.~!@#$%^&*()=+[{]}:?,<>/ 로 한정됩니다.

 

나의 접근방법 및 풀이

처음에는 for() 문을 사용해 for()문 안에서 한 번에 해결하는 문제인 줄 알았다. 하지만 문제를 다시 읽어보고 예시문을 읽어보니, '단계'라는 말처럼 차례대로 수행하면 될 것이라고 생각했다. 

 

●1단계 : 대문자 → 소문자

: 1단계에서는 별다른 어려움 없이 lower()를 이용하여 대문자를 소문자로 변환하였다.

## 1단계: 대문자-> 소문자
    new_id = new_id.lower()

 

●2단계 : 알파벳 소문자, 숫자, 빼기(-), 밑줄(_), 마침표(.)를 제외한 모든 문자를 제거

: 알파벳 '소문자'는 이전 단계에서 lower()를 통해 해결하였지만, '알파벳', '숫자', 특수문자 3종을 어떻게 해결하지 잠깐 고민했다.

for()문을 돌려 문자 하나하나를 체크하기로 하였고, if문을 이용하여 조건에 해당되는 문자는 answer에 추가하는 방식으로 진행하였다.

소문자는 isalpha(), 숫자는 isalnum(), 그리고 특수문자의 경우에는 리스트를 활용하여 구별하였다.

## 2단계: 소문자, 숫자, 빼기, ['-', '_', '.'] 이외 문자 제거
    for s in new_id:
        if s.isalpha() or s.isalnum() or s in ['-', '_', '.']: 
            answer += s

 

●3단계 : 마침표(.)가 2번 이상 연속된 부분 → 하나의 마침표(.)로 치환

: 3단계에서 엄청 난항을 겪었다. 연속되는 마침표 두 개를 하나로 바꾸는 게 왜?라고 할 수 있겠지만 우선 문제는 알겠는데 어떻게 구현해될지 생각이 안 났다. 그래서 찬스 느낌으로(?) 예전에 눈여겨봤던 같은 숫자는 싫어 문제를 풀어서 힌트를 얻었다. 

우선 이번 단계 역시 조건에 해당되면 미리 만들어둔 문자열에 추가하는 방식을 택했다.

문자열이 '.'이 아닐 경우와 '.' 이더라도 새로운 문자열의 마지막 문자와 같지 않으면 해당 문자를 추가했다.

## 3단계: 2번이상 연속되는 마침표 -> 하나의 마침표로
    tmp = ''
    for i in answer:
        if i != '.':    ## '.'이 아닐 경우
            tmp += i
        else:           ## '.' 일 경우
            if i != tmp[-1:]:
                tmp +=  i

 

●4단계 : 문자의 처음과 끝의 마침표('.') 제거

: 이번 단계에서는 슬라이싱을 이용했다. 

먼저 문자의 처음 또는  끝이 마침표('.')인 경우로 1차 판단하고 그다음에는 문자열의 길이에 따라 케이스를 나누었다.

여러 번 테스트해본 결과, 만약 tmp = '.'인 경우에는 tmp [0]이 tmp [-1] 이기 때문에 범위 오류가 발생했다. 그래서 문자열의 길이가 1인 경우, 즉, tmp = '.'인 경우에는 바로 공백 처리를 해주었다.

그다음, if문을 활용하여 첫 문자가 마침표인 경우, 마지막 문자가 마침표인 경우로 진행했다.

여기서! 마지막 문자를 판단하는 조건문을 elif로 작성하면 오류가 발생하니 if로 작성해야 한다.

## 4단계: 문자의 처음, 끝에 마침표 제거
    if tmp[0] == '.' or tmp[-1]=='.':    ## 처음
        if len(tmp) == 1:
            tmp = ''
        else:
            if tmp[0] =='.':
                tmp = tmp[1:]
            if tmp[-1] == '.':
                tmp = tmp[:-1]

 

●5단계 : 빈 문자열 → 'a'로 채우기

: 처음에는 replace()를 사용했지만, 적용이 되는 테스트가 있었고 안 되는 케이스가 있었다. 그래서 다시 생각해보니 문자열 중간에는 공백이 존재하는 경우가 아니니 바로 'a'를 채워도 된다고 판단하여, tmp += 'a'로 했다.

## 5단계: 빈 문자열은 'a'로
    if len(tmp) == 0:
        tmp += 'a'

 

 

●6단계 : 아이디 길이 (16자 이상) → 15자로 만들기

: 아이디의 최대 길이가 15자 이기 때문에 이를 해결하는 것은 쉬웠다. 하지만 아이디가 15번째 문자가 마침표('.') 일 경우에는 한 번 더 처리를 해줘야 하는 조건이 있어 슬라이싱을 이용했다.

## 6단계: 아이디 길이 (16자 이상)
    if len(tmp) >=16:
        tmp = tmp[:15]
        #print(tmp)
        if tmp[-1] == '.':
            tmp = tmp[:-1]

 

●7단계 : 아이디 길이 (2자 이하) → 마지막 글자로 3자 이상으로 만들기

: while() 문을 사용하여 마지막 문자를 tmp에 반복하여 더했다.

 ## 7단계 아이디 길이 (2자 이하)
    if len(tmp) <= 2:
        while(len(tmp) != 3):
            tmp += tmp[-1]

 

전체 코드

def solution(new_id):
    answer = ''
    ## 1단계: 대문자-> 소문자
    new_id = new_id.lower()

    ## 2단계: 소문자, 숫자, 빼기, ['-', '_', '.'] 이외 문자 제거
    for s in new_id:
        if s.isalpha() or s.isalnum() or s in ['-', '_', '.']: 
            answer += s

    ## 3단계: 2번이상 연속되는 마침표 -> 하나의 마침표로
    tmp = ''
    for i in answer:
        if i != '.':    ## '.'이 아닐 경우
            tmp += i
        else:           ## '.' 일 경우
            if i != tmp[-1:]:
                tmp +=  i          
    
    ## 4단계: 문자의 처음, 끝에 마침표 제거
    if tmp[0] == '.' or tmp[-1]=='.':    ## 처음
        if len(tmp) == 1:
            tmp = ''
        else:
            if tmp[0] =='.':
                tmp = tmp[1:]
            if tmp[-1] == '.':
                tmp = tmp[:-1]
    
    ## 5단계: 빈 문자열은 'a'로
    if len(tmp) == 0:
        tmp += 'a'
        
    ## 6단계: 아이디 길이 (16자 이상)
    if len(tmp) >=16:
        tmp = tmp[:15]
        #print(tmp)
        if tmp[-1] == '.':
            tmp = tmp[:-1]

    ## 7단계 아이디 길이 (2자 이하)
    if len(tmp) <= 2:
        while(len(tmp) != 3):
            tmp += tmp[-1]
        
    return tmp

print(solution("...!@BaT#*..y.abcdefghijklm"))
print(solution("=.="))

 

 

다른 풀이

def solution(new_id):
    answer = ''
    # 1
    new_id = new_id.lower()
    # 2
    for c in new_id:
        if c.isalpha() or c.isdigit() or c in ['-', '_', '.']:
            answer += c
    # 3
    while '..' in answer:
        answer = answer.replace('..', '.')
    # 4
    if answer[0] == '.':
        answer = answer[1:] if len(answer) > 1 else '.'
    if answer[-1] == '.':
        answer = answer[:-1]
    # 5
    if answer == '':
        answer = 'a'
    # 6
    if len(answer) > 15:
        answer = answer[:15]
        if answer[-1] == '.':
            answer = answer[:-1]
    # 7
    while len(answer) < 3:
        answer += answer[-1]
    return answer

우선 3단계를 별다른 if문 없이 while() 문과 다른 문자열에 추가 없이 replace()를 이용한 점이 좋았다. 

4단계도 굳이 첫 번째와 마지막을 여러 번 분리하지 않고 구현해서 좋았다.

7단계에서도 if() 문을 사용하지 않고, while() 문의 조건으로 문자열의 길이를 사용할 수 있다는 것을 다시 깨달았다.

 

 

 

 

 

사실 다 풀고 다시 코드만 보니 그렇게 어렵게 느껴지는 코드는 아니지만, 여전히 문제가 맨땅에 헤딩으로 느껴지던 나는 무려 5시간이나 걸려 풀었던 문제이다.(물론 설렁설렁하게 풀었지만)

끝까지 풀어냈다는 뿌듯함과 성취감이 느껴지는 한편 약간의 자괴감도 느껴지는 문제였다.

그래도 계속 풀어나가다 보면 언젠가는 착착착! 풀어나가는 그날을 위해 더욱더 열심히 해야겠다고 다짐하게 되는 문제였다.