[프로그래머스] 둘만의 암호 - 파이썬

링크: 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

---

입출력 예

s	skip	index	result
"aukks"	"wbqd"	5	"happy"

---

문제풀이

이 문제를 보고 전에 풀었던 프로그래머스의 시저 암호가 생각나서 같은 방법으로 풀어줬다.

시저암호: https://velog.io/@ehdtkd98/프로그래머스시저-암호-파이썬

문제 이해하기:

두 문자열 s, skip 이 주어지고 자연수 index 가 주어진다. 문자열 s 의 각 알파벳을 index 만큼 뒤의 알파벳으로 바꿔줘야 하는데, 이 과정에서 skip 에 있는 알파벳은 건너뛰어야 한다.

목표: 변환 결과 return 하기

내 풀이:

• s 의 각 문자열을 돌면서 while 문을 사용한다.
  • index 보다 작을 때
    • 문자열 char 를 chr, ord 를 활용해 + 1을 해서 바꿔준다.
    • 만약 바꾼 문자열 char 가 z 보다 크다면 → char = ‘a’
    • 만약 char 가 skip 리스트에 없으면? → 횟수 올리기
  • while 문을 다 돌렸으면 변환이 완료된 char 값을 answer 에 더하기

최종코드

def solution(s, skip, index):
    answer = ''
    
    for char in s:  # s의 각 문자에 대해
        cnt = 0
        while cnt < index:  # index 보다 작을 때
            char = chr(ord(char) + 1)  # 문자를 다음 문자로 바꾸고
            if ord(char) > 122:  # 만약 바꿨을 때 z의 아스키코드를 넘어버린다면
                char = 'a'  # a로 바꾸기.
            if char not in skip:  # skip 리스트에 없으면 횟수 늘리고
                cnt += 1  
        
        answer += char  # while 문을 다 돌았다면 변환 완료된 문자열을 answer에 추가.

    return answer