문제 설명
두 문자열 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" |
내 풀이
def solution(s, skip, index):
answer = ''
# skip을 제외한 알파벳 리스트를 생성
alpha = [chr(i) for i in range(ord('a'),ord('z')+1) if not chr(i) in list(skip)]
# s를 순회
for j in s:
# alpha에서 해당하는 인덱스를 찾아와 index만큼 뒤의 알파벳을 찾음
# 알파벳이 z를 넘어갈 경우 다시 a로 돌아가기 때문에 alpha의 길이만큼 나눈 나머지를 이용
idx = (alpha.index(j) + index) % len(alpha)
answer += alpha[idx]
- 리스트 컴프리헨션을 이용해 skip을 제외한 알파벳 리스트를 생성(if 문을 사용)