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

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

프로그래머스

문제 설명

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

문제 풀이

1. 먼저 스킵하는 문자의 경우 따로 기록합니다.
2. 그 뒤 문자열 s 안의 문자에 대해서 선형적으로 처리합니다.
   1. 아스키 코드로 변환한 후, 1씩 더해가며 z를 넘어갈 경우 a로 만듭니다.
   2. 만약 스킵해야 할 문자열일 경우 continue, 그 외의 경우 넘어간 것으로 계산하여 반복합니다.
3. 최종적으로 결과를 문자열로 바꿔 합칩니다.

def solution(s, skip, index):
    
    answer = ""
    
    skip_dict = dict()
    for skip_case in skip:
        skip_dict[ord(skip_case)] = True
        
    for letter in s:
        letter_code = ord(letter)
        convert_count = 0
        while convert_count < index:
            
            letter_code += 1
            
            if letter_code > ord("z"):
                letter_code = ord("a")
            
            if skip_dict.get(letter_code):
                continue
            else:
                convert_count += 1
        
        answer += chr(letter_code)
        
    return answer