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

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"

입출력 예 설명

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

풀이

설명

• 문자열을 아스키코드로 고쳐 풀이한다.
  • ord(), chr() 함수 사용
• index 값을 cnt에 할당하여 cnt가 0이 되기 전까지 word의 값을 하나씩 증가시킨다.
• 단, z보다 크거나 skip 하고자 하는 단어 리스트에 있는 경우 a로 돌아가거나 다음 알파벳으로 넘어간다.

코드

# 먼저 ord() 사용하여 문자열을 아스키코드 값으로 변경
# 마지막에 chr() 사용하여 다시 아스키코드 값을 문자열로 변경
def solution(s, skip, index):
    answer = ''
    skipList = list(ord(s) for s in skip)
    
    for word in s :
        cnt = index
        word = ord(word)
        
        while cnt != 0 :
            word += 1
            
            if word > ord('z') :
                word = word - ord('z') + ord('a') - 1
            
            if word in skipList :
                continue
            
            cnt -= 1
        answer += chr(word)
    
    # 제한사항
    if 5 > len(s) > 50 :
        print("s의 범위 초과")
    if 1 > len(skip) > 10 :
        print("skip의 범위 초과")
    if 1 > index > 20 :
        print("index의 범위 초과")
    if s == skip :
        print("s와 skip 문자열 중복")
    for i in s :
        if ord(i) < ord('Z') :
            print("s 소문자만 가능")
    for j in skip :
        if ord(j) < ord('Z') :
            print("skip 소문자만 가능")
        
    return answer