[python] 프로그래머스 Lv 1 - 둘 만의 암호

🔗 문제 링크

프로그래머스

📁 분류

구현

💡풀이

이 문제는 s에서 skip에 포함된 문자를 제외하고 index만큼 이동하여 새로운 문자열을 생성하는 문제이다.

 

문제를 풀 때 참고해야 할 점은 z를 넘어갈 경우 a로 돌아가는 것이다.

 

예를 들어 s가 z이고 skip이 ['abcde']이고 index가 1일 경우 z는 a로 돌아가기 때문에 a를 건너뛰고 b를 건너뛰어야 한다.

💻 코드

def solution(s, skip, index):
    answer = ''
    s = [ord(i) - ord('a') for i in s]
    skip = [ord(i) - ord('a') for i in skip]
    for c in s:
        count = 0
        i = 1
        while True:
            if (c + i) % 26 not in skip:
                count += 1
            if count == index:
                answer += chr(((c + i) % 26) + ord('a'))
                break
            i += 1

    return answer