TIL-40. [코딩테스트] 프로그래머스 lv1. 카카오 조건에 맞는 아이디 생성하기

solarrrrr·2022년 1월 8일
0

Today I Learned

목록 보기
40/74
post-thumbnail

아래 조건을 만족하는 결과를 도출해야 한다.

1단계 newid의 모든 대문자를 대응되는 소문자로 치환합니다.
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이 될 때까지 반복해서 끝에 붙입니다.

문제 보는 순간 여러 방법이 떠올랐지만
문자열 컨트롤하는 건 정규식이 가장 편해서 아래와 같이 풀어보았다.

import re

def solution(new_id):
  answer = new_id.lower() 
  answer = re.sub('[^a-z0-9\-\_\.]',"", answer) 
  answer = re.sub('\.+',".", answer) 
  answer = re.sub('^[.]|[.]$', "", answer)  
  answer = re.sub('^$',"a", answer)
  answer = answer[:15] 
  answer = re.sub('^[.]|[.]$', "", answer) 

  if len(answer) < 3:
    add_str = answer[-1]
    while 3-len(answer):
      answer = answer + add_str
    
  return answer

처음엔 정규식 1~2줄로 끝내고 싶어서 이리저리 해 봤는데 잘 안 됐다.
오토핫키에서는 중복 조건을 걸어서 잘 됐던 거 같은데
파이썬에서는 함수 내에서 한 줄씩 순차 처리가 된다.
그래서 문자열의 처음과 끝에 마침표가 들어가면 삭제해 주는 정규식도
15자로 글자를 자른 다음 라인에 별도로 넣어줬다.
이렇게 안 하면 문제 통과를 못 한다.

루프를 넣어볼까도 싶었는데 그게 더 비효율일 거 같아서
시키는 대로 단계별로 작성했다.

아래는 가장 많은 추천(?) 받은 사람들의 코드다.


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

정규식을 통한 접근 방식과 마무리까지 거의 나랑 비슷하게 푼 거 같다.
다만 이분들은 삼항연산자로 한 줄에 처리되도록 깔끔하게 풀었다.

profile
몰입

0개의 댓글