[프로그래머스 Lv.1] - 둘만의 암호

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 함수를 완성해주세요.

 

 

구현

def solution(s, skip, index):
    skipNum = [ord(i) for i in skip]
    answer = ''
    for i in s:
        num = ord(i)
        count= 0
        for j in range(num+1, num+index+1):
            if j in skipNum:
                count = count + 1
		
        total = num+index+count
        resultNum = total-26 if total > ord('z') else total
        answer = answer + chr(resultNum)
    return answer

 

처음에는 이런 방식으로 구현하였더니 1번째 테스크테이스를 제외하고는 모두 실패하였다. 논리대로 풀었는데 90% 실패율은 처참하였다. 위 코드는 순환이 불가능하고 일회성만 가능한 코드로 인지하여 다음과 같은 방법으로 명시적으로 해결해나갔다. 

 

def solution(s, skip, index):
    alpha = 'abcdefghijklmnopqrstuvwxyz'
    answer = ''
    for i in skip:
        alpha = alpha.replace(i, '')
    for a in s:
        resultNum = (alpha.find(a) + index) % len(alpha)
        answer = answer + alpha[resultNum]
    return answer

 

skip에 지정된 문자를 최상단에서 미리 제거하는 방법을 생각해냈다면 쉽게 풀 수 있는 문제였다.