알고리즘

[Python] 카카오 2021 BLIND : 신규 아이디 추천

ji_iin 2021. 9. 13. 13:50

문제

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

[ 문제 조건 ]

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. str.lower() : 대→소문자 치환
  2. re.sub( 패턴, 대체 문자, 바꿀 문자열) *(^ : 시작)
  3. \.+ : .을 특수문자로 인식하고 한 개 이상일 때(+)
  4. str.strip('.') : 양 끝의 . 을 제거
  5. 빈 문자열 일 때 a 대입
  6. [0:15] : 문자열의 0~15까지 slice
    1. [-1] : 문자열의 마지막 문자
    2. str.rstrip('.') : 맨 오른쪽 . 삭제
  7. 문자열에 문자 추가

✏️한줄평

: 여태까지는 알고리즘에 매번 나오는 유형만 풀다가 정규식, 문자열 다루는 문제를 푸니 색달랐다. 대기업 코테 기출도 자주 풀어야겠다.


코드

import re
def solution(new_id):
new_id = new_id.lower() #1단계
new_id = re.sub('[^0-9a-z-_.]','',new_id) #2단계
new_id = re.sub('\.+','.',new_id) #3단계
# \ : 문자 앞 슬래쉬는 특수문자로 인식하고 문자 그대로 해석하지 X
# + : 1개 이상
#4단계
new_id = new_id.strip('.')
#5단계
if new_id == "":
new_id = 'a'
#6단계
if len(new_id) > 15:
new_id = new_id[0:15]
if new_id[-1] == '.':
new_id = new_id.rstrip('.')
#7단계
if len(new_id) <= 2:
c = new_id[-1]
while len(new_id)!=3:
new_id += c
return new_id
print(solution("z-+.^."))
view raw kakao_2021-1.py hosted with ❤ by GitHub

참고하면 좋은 코드

import re

def solution(new_id):
    st = new_id
    st = st.lower()
    st = re.sub('[^a-z0-9\-_.]', '', st)
    st = re.sub('\.+', '.', st)
    st = re.sub('^[.]|[.]$', '', st)
    st = 'a' if len(st) == 0 else st[:15]
    st = re.sub('^[.]|[.]$', '', st)
    st = st if len(st) > 2 else st + "".join([st[-1] for i in range(3-len(st))])
    return st