[프로그래머스 Python] Lv 1. 둘만의 암호

---

문제 설명

두 문자열 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) 코드 작성

def solution(s, skip, index):
    alpha = [a for a in 'abcdefghijklmnopqrstuvwxyz'*3 if a not in skip]
    result = ''
    for ss in s:
        i = alpha.index(ss)
        result += alpha[i+index]
    return result

(2) 코드 리뷰

alpha라는 리스트에 skip에 포함되는 알파벳을 제외한 나머지 알파벳을 담아주었다. 이때 *3을 한 이유는 제한사항 때문이다. 제한사항을 보면 skip이 최대 10, index 최대 20이다. 영어 알파벳 총 26개 중에 10개가 잘려나갈 수 있다. 만약 10개의 skip 항목이 들어온다면 남은 알파벳은 16개이고 index 값이 20으로 들어온다면 인덱스 range를 벗어나게 된다. 마지막 16번째에 있는 값에서 20만큼 뒤에 있는 알파벳을 불러오기 위해선 알파벳의 길이는 최소 36개로 맞춰줘야 하기 때문에 *3을 해준 것이다.

👩🏻‍💻개인 공부 기록용 블로그입니다
오류나 틀린 부분이 있을 경우 댓글 혹은 메일로 따끔하게 지적해주시면 감사하겠습니다.