Python/알고리즘 문제 풀이

[코테] 신규 아이디 추천

마이구미+ 2023. 7. 18. 10:49

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

문제링크

 

프로그래머스

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

programmers.co.kr

<로직 고민/>

  • if문을 주르륵 쓰라고 낸 문제는 아니겠지?
  • 일단 하나씩 차근차근 단계별로 해봤다
import string

def solution(new_id):
    new_id_len = len(new_id)
    
    for id in new_id:
        # 1단계
        if id.isupper():
            new_id = new_id.replace(id, id.lower())

        # 2단계
        special_char = string.punctuation.replace("-", "").replace("_", "").replace(".", "")
        for s in special_char:
            new_id = new_id.replace(s, "")

        # 3단계
        new_id = new_id.replace("..", ".")

        # 4단계
        new_id = new_id.strip(".")
        
        # 5단계
        if new_id == "":
            new_id += "a" * new_id_len
        
        # 6단계
        if len(new_id) >= 16:
            new_id = new_id[:15]
            
        # 7단계
        if len(new_id) <= 2:
            final_char = new_id[-1]
            while len(new_id) < 3:
                new_id += final_char
                
    return new_id
  • 이렇게 해서 예시는 다 통과하는데 채점하니까 테스트 16,17,20,21,22,23에서 실패한다
  • 음 좀 걸리는 게 1단계랑 3단계 말고는 for문이 필요없다는 거..
  • 아니 1단계도 필요없을 듯
  • 처음에 그냥 new_id.lower() 했는데 안 돼서 저렇게 한 건데 
  • new_id = new_id.lower() 하면 될 듯
...
    # 1단계
    new_id = new_id.lower()
...
    # 3단계
    for _ in range(len(new_id)):
        new_id = new_id.replace("..", ".")
...
  • 6번째 줄에 있던 for문을 없애고 1단계랑 3단계를 수정했다
  • 이렇게 했더니 이제 테스트 22,23만 실패한다
  • 아 대박 5단계가 문제였다
  • new_id가 빈문자열이면 new_id의 원래 길이만큼 a를 더하는 게 아니라 그냥 a를 대입하는 거였다,,,;;
# 5단계
if new_id == "":
    new_id = "a"
  • 아 지금 보니 특수문자 제한사항이 있었네 string import 할 필요 없었네...

<완성된 풀이/>

import string

def solution(new_id):
    # 1단계
    new_id = new_id.lower()

    # 2단계
    special_char = string.punctuation.replace("-", "").replace("_", "").replace(".", "")
    for s in special_char:
        new_id = new_id.replace(s, "")

    # 3단계
    for _ in range(len(new_id)):
        new_id = new_id.replace("..", ".")

    # 4단계
    new_id = new_id.strip(".")

    # 5단계
    if new_id == "":
        new_id = "a"

    # 6단계
    if len(new_id) >= 16:
        new_id = new_id[:15]
        if new_id[-1] == ".":
            new_id = new_id.rstrip(".")

    # 7단계
    if len(new_id) <= 2:
        final_char = new_id[-1]
        while len(new_id) < 3:
            new_id += final_char
                
    return new_id