프로그래머스[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

 

💁입출력 예

s	skip	index	result
"aukks"	"wbqd"	5	"happy"

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

 

🙆코드

def solution(s, skip, index):
    alphabet = 'abcdefghijklmnopqrstuvwxyz' * 2

    for i in skip:
        alphabet = alphabet.replace(i, "")

    answer = ''
    for alpha in s:
        try:
            idx = alphabet.index(alpha)
            new_idx = (idx + index) % len(alphabet)
            answer += alphabet[new_idx]
        except ValueError:
            answer += alpha

    return answer

 

🗨️첨언

alphabet 변수에 알파벳 길이를 두배로 저장한다.

skip에 해당하는 문자들은 제거한다.

alphabet에서 alpha의 인덱스 idx 를 찾는다.

alpha가 alphabet에 없으면 valueError가 발생할 수 있으므로 예외처리한다.

idx+ index % len(alphabet)을 계산하여 새로운 인덱스를 구한다.