Daily IT Study

(출처: 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 함수를 완성해주세요.

제한사항

• 5 ≤ s의 길이 ≤ 50

• 1 ≤ skip의 길이 ≤ 10

• s와 skip은 알파벳 소문자로만 이루어져 있습니다.
  • skip에 포함되는 알파벳은 s에 포함되지 않습니다.

• 1 ≤ index ≤ 20

입출력 예

입출력 예 설명

입출력 예 #1본문 내용과 일치합니다.

풀이

• 문자열 의 각 알파벳을 index만큼 뒤 알파뱃

• z를 넘을 경우 a로 돌아감

• skip에 있는 알파벳을 제외

1. 파이썬에서 알파벳을 아스키코드로 변환

2. skip으로 들어온 값들 제외

3. a부터 정렬하여 딕셔너리로 저장

!인덱스 지정 방법의 멋진 코드

dic_alpha = {alpha:idx for idx, alpha in enumerate(a_to_z)}
print(dic_alpha)
//{'a': 0, 'c': 1, 'e': 2, 'f': 3, 'g': 4, 'h': 5, 'i': 6, 'j': 7, 'k': 8, 'l': 9, 'm': 10, 'n': 11, 'o': 12, 'p': 13, 'r': 14, 's': 15, 't': 16, 'u': 17, 'v': 18, 'x': 19, 'y': 20, 'z': 21}

정답

from string import ascii_lowercase
def solution(s, skip, index):
    answer = ''
    a_to_z = set(ascii_lowercase)
    a_to_z = sorted(a_to_z - set(skip))
    l = len(a_to_z)
    # print(l)

    dic_alpha = {alpha:idx for idx, alpha in enumerate(a_to_z)}

    for i in s:
        answer += a_to_z[(dic_alpha[i] + index) %l]
    return answer