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

문제 설명

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

 

https://school.programmers.co.kr/learn/courses/30/lessons/155652

프로그래머스

제한 사항

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 문자열의 각 문자를 0~25의 정수로 변환하여 리스트로 저장 (a=0, z=25)
    skips = [ord(char) - 97 for char in skip]  
    
    for char in s:  
    	# 현재 문자를 0~25의 정수로 변환 (a=0, z=25)
        current = ord(char) - 97  
        # 현재 몇번 이동했는지 저장
        steps = 0  
        
        # 지정된 index만큼 이동할 때까지 반복
        while steps < index:  
        	# 현재 문자 값을 1 증가 (z를 넘어가면 다시 a로 순환)
            current = (current + 1) % 26  
            
            # skips 리스트에 없는 문자만 유효한 이동으로 간주
            if current not in skips:  
                steps += 1  
            
        # 이동 완료한 값을 다시 문자로 변환하여 결과 문자열에 추가   
        answer += chr(current + 97)  
    
    return answer

---

파이썬을 독학하시는 분들에게 도움이 되길 바라며,
혹 더 좋은 방법이 있거나 오류가 있다면 편하게 말씀 부탁드립니다.