둘만의 암호

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

 

[풀이]

아스키 코드를 이용해서 문제를 풀면 되겠구나 생각했고, 아스키 코드로 변환 후 index만큼 더하는데 소문자 알파벳 범위를 넘어가면 안되니까 그 부분을 모듈 연산자로 처리를 해준다. 근데 skip 에 있는 알파벳은 건너뛰어야 하니까 이 부분을 반복문으로 처리하면 되겠다 싶었다. 1을 더한 값이 skip에 있는지 없는지 확인을 해야하니 좀 더 효율적으로 처리하기 위해 자료구조를 set으로 하였다.

def solution(s, skip, index):
    answer = []
    # 효율적인 처리를 위해 set으로 해서 O(1)로 만들어줌
    skip = set(map(ord, skip))

    for c in map(ord, s):
        step = 0
        
        # 스텝이 index보다 작으면 반복
        while step < index:
        
        	# c를 한칸 이동함
            c = (c + 1 - ord('a')) % 26 + ord('a')
            
            # 이동한 것이 skip에 있는가?
            if c not in skip:
            
            	# 없다면 스텝 1 증가
                step += 1
                
        # 작업이 끝났다면 리스트에 추가
        answer.append(chr(c))

    return ''.join(answer)