[프로그래머스] 둘만의 암호 | 파이썬

문제 설명

두 문자열 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 함수를 완성해주세요.

 

설명

문제의 요구사항은 암호화된 문자열과 자연수가 주어졌을때 각 문자의 자연수 만큼 뒤에 있는 문자로 변환하라는 것이다. 이때 그냥 뒤에 있는 문자라고 하면 심심하니 또 새로운 문자열 skip이 주어지면서 이에 해당하는 문자열을 제외하고 넘어가는 숫자를 세어야한다. 처음에는 skip문자열 처리를 어떻게 해야하나 고민을 많이 했으나 생각보다 간단하게 해결 할 수 있었다. 그것은 풀이의 alpha와 같이 알파벳을 순서대로 나열하고 skip에 해당하는 알파벳을 제거한 새로운 순서를 만들어주는 것이다. 그러면 skip에 해당하는 부분이 고려되어 암호를 해독할 수 있다.

 

풀이

def solution(s, skip, index):
    answer = ""
    alpha = "abcdefghijklmnopqrstuvwxyz"
    for i in list(skip):
        alpha = alpha.replace(i,"")
    for a in s:
        answer += alpha[(alpha.find(a) + index) % len(alpha)]
    return answer