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

link 🔗 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

 

---

 

def solution(s, skip, index):
    answer = ''
    # skip 여부 기록 리스트
    check = [1] * ((ord('z') - ord('a')) + 1)
    last = len(check)
    
    # skip 알파벳 기록
    for skipAlp in skip:
        check[ord(skipAlp) - ord('a')] = 0
    
    for alp in s:
        idx = 1
        # skip에 없는 알파벳 건너 뛴 횟수
        passIdx = 0
        change = ''
        alpOrd = ord(alp) - ord('a')
        # index 이상 건너뛰면 종료
        while passIdx < index:
            checkOrd = alpOrd + idx
            # 현재 위치가 리스트 벗어나면 (z를 넘어가면)
            if checkOrd >= last:
                # 다시 a로 복귀
                checkOrd %= last
            
            # 건너 뛸 수 있으면 (skip에 해당 알파벳 없으면)
            if check[checkOrd]:
                # pass
                passIdx += 1
                # pass한 횟수가 인덱스와 같으면 => answer에 알파벳 더하기
                if passIdx == index:
                    change = chr(checkOrd + ord('a'))
            # 다음 알파벳으로
            idx += 1
            
        answer += change
        
    return answer

 

✏️ s의 각 알파벳을 index개 뒤에 있는 것으로 교체해줘야 하는데, 만약 중간에 skip에 들어 있는 알파벳이라면 건너뛰어야 한다. 이 점이 로직을 짜는데 조금 헷갈렸다.

✏️ idx는 skip에 알파벳이 있든 없든, 일단 다음 칸으로 가기 위한 변수이다.

✏️ passIdx라는 변수를 만들어서, 만약 이 변수의 값이 index와 같아진다면 a보다 skip이 반영되면서 index개 뒤에 있는 알파벳을 찾은 것이므로 answer에 더해준다.

✏️ 일부 문제에서 인덱스 에러가 났는데, 위치가 'z'를 넘어갈 경우 다시 'a'로 복귀해주는 과정에서 문제가 있었다. 처음에는 단순히 26(z의 순번)을 빼줬는데, 몇 바퀴를 돌 수도 있다는 걸 간과했다! 그래서 26을 빼는 것이 아니라, 26으로 나눈 나머지를 사용하는 것으로 로직을 수정했다.