[코딩테스트] 프로그래머스 둘만의 암호 파이썬(Python)

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

프로그래머스

def solution(s, skip, index):
    alphabet = [chr(i) for i in range(ord('a'), ord('z')+1) if chr(i) not in skip]
    alphabet_to_index = {}
    index_to_alphabet = {}
    for i, c in enumerate(alphabet):
        alphabet_to_index[c] = i
        index_to_alphabet[i] = c
    result = ''
    for c in list(s):
        i = (alphabet_to_index[c] + index) % len(alphabet)
        result += index_to_alphabet[i]
    return result

먼저 list comprehension과 조건문을 활용하여 skip에 존재하지 않은 알파벳만 구하여 alphabet 리스트로 만들었습니다.
그리고 enumerate()함수를 사용하여 인덱스 i와 c를 alphabet리스트에서 순차적으로 추출합니다.
alphabet_to_index, index_to_alphabet 딕셔너리에는 다음과 같은 형태로 저장됩니다.

• alphabet_to_index : {알파벳 : 인덱스}
• index_to_alphabet : {인덱스 : 알파벳}

s를 반복하면서 i를 구합니다.

• i는 alphbet_to_index에서 c에 해당하는 인덱스에 index변수를 더하고 len(alphabet)으로 나눈 나머지입니다.
• i는 index만큼의 뒤의 알파벳이 z를 넘어갈 경우 다시 a로 돌아갈 수 있게 해줍니다.

위에서 구한 i를 index_to_alphabet에서 추출하여 result변수에 저장합니다.

반복문을 종료한후에 result변수를 반환합니다.