[Programmers] Level 1 - 둘만의 암호

문제

프로그래머스

 

풀이

def solution(s, skip, index):
    # 초기화
    answer = ''

    for alpha in s:
        origin = ord(alpha)
        
        # index까지 한 칸 씩 이동
        for _ in range(1, index + 1):
            origin += 1

            # z를 넘어가면 a로 다시 돌아오는 조건
            if origin > ord('z'):
                origin -= 26

            # skip 문자열이 포함되는 경우 한 칸 이동
            while chr(origin) in skip:
                origin += 1

                # z를 넘어가면 a로 다시 돌아오는 조건
                if origin > ord('z'):
                    origin -= 26

        answer += chr(origin)

    return answer

 

채점 결과