[프로그래머스] LV.1 둘만의 암호 / 파이썬(Python)

 

📜 문제 설명

두 문자열 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

 

🔗 문제 링크

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

프로그래머스

 

💡 문제 풀이

아스키 코드를 활용해서 풀어본 문제

• 아스키 코드로 변환하면 소문자 a-z의 범위는 97에서 122이다.
  122가 넘으면 97로 초기화한다. 즉 문자열이 z인데 더 더해주어야 하면 a로 초기화 한다.
• 한 글자씩 숫자 형태로 변환하고 index 만큼 더해준다.
  이 때 중간에 skip에 해당하는 문자열이 있으면 1씩 더 더해준다.
• 위의 작업이 완료되면 다시 문자화하여 answer에 더해준다.

def solution(s, skip, index):
    answer = ''
    skip_list = []
    
    # 건너 뛰어야 하는 알파벳의 목록을 숫자로 정리한다.
    for i in range(len(skip)):
        skip_list.append(ord(skip[i]))
    
    # 문자열 s를 한 글자씩 순회하면서
    # 정수로 변환한다
    for i in range(len(s)):
        tmp = ord(s[i])
        tmp_idx = index
       
       # 문자가 z를 지나치면 a로 초기화하고
       # skip 대상인 문자열을 제외하고는 한 글자씩 더해준다
        while tmp_idx:
            tmp += 1
            if tmp > 122:
                tmp = 97
            if tmp not in skip_list:
                tmp_idx -= 1
        
        # 다시 문자열로 변환해서 answer에 더해준다
        answer += chr(tmp)
        
    return answer