[코딩테스트 연습] Programmers 코딩테스트 연습 > 연습문제 > 둘만의 암호 - python

[Programmers] 둘만의 암호 (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

입출력 예

sskipindexresult

"aukks"

"wbqd"

5

"happy"

 

💡 문제 풀이

해당 문제는 python 내장함수인 ord()와 chr()를 이용하여 문제를 해결하였습니다.

 

- ord() Python 3.11.1 문서 : # https://docs.python.org/ko/3/library/functions.html?highlight=ord#ord 

Built-in Functions

- chr() Python 3.11.1 문서 : # https://docs.python.org/ko/3/library/functions.html?highlight=ord#chr 

Built-in Functions

Tip. 해당 문제에서 아래 조건을 처리하는게 가장 중요한 내용이라고 생각합니다.

그래서 저는 while 문 안에서 문자열이 'z' 다음이 될 경우에 'a'로 치환해주었습니다.

• index만큼의 뒤의 알파벳이 z를 넘어갈 경우 다시 a로 돌아갑니다.

#Python 코드

def solution(s, skip, index):
    result = ''
    arr = [ord(i) for i in skip]
    
    for char in s:
        cnt = 0
        while(cnt < index):
            char = chr((ord(char)+1))

            if ord(char) > ord('z'):
                   char = 'a'
                
            if ord(char) not in arr:
                cnt += 1
                            
        result += char
        
    return result

 

출처: 프로그래머스 코딩 테스트 연습, https://school.programmers.co.kr/learn/challenges