[4코1파] #45. 둘만의 암호

gunny·2023년 2월 16일

4코1파 코딩테스트 코딩테스트 python 프로그래머스 둘만의 암호 프로그래머스 둘만의 암호 python

코딩테스트

목록 보기

45/530

[4코1파] 4명의 안드로이드 개발자와 1명의 파이썬 개발자의 코딩 테스트 서막 : 4코1파

Rule :

하루에 1문제씩 풀기.
한 문제당 30분씩은 고민하기.
왜 그렇게 풀었는지 공유하기.
하루라도 놓친다면 벌금은 1,000원

START :

[3코1파] 2023.01.04~ (45일차)
[4코1파] 2023.01.13~ (36일차)

Today :

2023.02.17 [45일차]

프로그래머스 LV1
둘만의 암호
https://school.programmers.co.kr/learn/courses/30/lessons/155652

문제 설명
두 문자열 s와 skip, 그리고 자연수 index가 주어질 때, 다음 규칙에 따라 문자열을 만들려 합니다. 암호의 규칙은 다음과 같습니다.

문자열 s의 각 알파벳을 index만큼 뒤의 알파벳으로 바꿔줍니다.
index만큼의 뒤의 알파벳이 z를 넘어갈 경우 다시 a로 돌아갑니다.
skip에 있는 알파벳은 제외하고 건너뜁니다.
예를 들어 s = "aukks", skip = "wbqd", index = 5일 때, a에서 5만큼 뒤에 있는 알파벳은 f지만 [b, c, d, e, f]에서 'b'와 'd'는 skip에 포함되므로 세지 않습니다. 따라서 'b', 'd'를 제외하고 'a'에서 5만큼 뒤에 있는 알파벳은 [c, e, f, g, h] 순서에 의해 'h'가 됩니다. 나머지 "ukks" 또한 위 규칙대로 바꾸면 "appy"가 되며 결과는 "happy"가 됩니다.

두 문자열 s와 skip, 그리고 자연수 index가 매개변수로 주어질 때 위 규칙대로 s를 변환한 결과를 return하도록 solution 함수를 완성해주세요.

제한사항

5 ≤ s의 길이 ≤ 50
1 ≤ skip의 길이 ≤ 10
s와 skip은 알파벳 소문자로만 이루어져 있습니다.
skip에 포함되는 알파벳은 s에 포함되지 않습니다.
1 ≤ index ≤ 20

입출력 예

입출력 예 설명

입출력 예 #1
본문 내용과 일치합니다.

문제 풀이 방법

string 모듈을 이용해서 아스키코드로 lower alpha를 불러옴
lower alpha 문자열에서 skip 해야 하는 단어를 정규표현식으로 제거
s 문자열을 하나씩 돌면서 index 값을 더해줌
index 값을 더해줬을 대, 소거된 alpha 길이(인덱스)보다 클 경우에는 값을 빼주면서 문자열을 찾아감

처음에는 그냥 문자열을 한 번만 빼주는 로직으로 했더니
테스트 케이스,
3,17,18,19 박살

idx의 범위가 1이상 20이하 일때,
만약 skip 해야 하는 단어의 개수가 10개라면, 위 로직에서 idx 에서 나머지 len(alpha)를 뺐을 때
19가 나와서, alpha의 하한선인 15보다 커서 인덱싱 오류가 나게 된다~
그래서, len 을 뺄 때 몫을 이용해서 두번, 세번 뺄 수 있게 수정
했음

그롸치

내 코드

import string, re

def solution(s, skip, index):
    answer = ''
    alpha = string.ascii_lowercase
    
    for s1 in skip:
        alpha = re.sub(s1, '',alpha)
        
    for s2 in s:
        idx = alpha.find(s2)+index
        if idx >= len(alpha):
            idx = idx-(len(alpha)*(idx//len(alpha)))            
        answer+= alpha[idx]
    
    return answer

증빙