[Programmers] 연습문제 > 둘만의 암호

문제 설명

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

입출력 예

"aukks"

"wbqd"

5

"happy"

---

[풀이]

문자를 숫자로 변환해서 쓰기 위해 ord(), chr() 메소드로 변환했고

a~z 가 97~122 여서

s에 있는 한 글자씩 skip과 겹치지 않는 걸 총 index까지 세는게 cnt고,

skip과 겹치는 거까지 포함해서 index까지 도달하도록 세는걸 r_cnt로 함.

 

def solution(s, skip, index):
    answer = '' #97~122
    for word in s:
        cnt = 0 # 겹치지 않는 단어를 index까지 세기
        r_cnt = 0 # 겹치는 단어까지 포함해서 얼마나 지나야 skip에 포함되지 않는지 index까지 세기
        while(cnt!=index):
            num = ord(word) + r_cnt # 현재 word + r_cnt 해서 index를 찾기 위해 지나가는 현재 단어 위치
            while(num>=122): # 최대 50넘어갈 수 있어서 2번 이상 a~z 반복 할 수 있음
                num-=26
            if chr(num+1) not in skip:
                cnt += 1
            r_cnt += 1
        final = ord(word)+r_cnt # 마지막 구한 r_cnt 까지 더해서 파이널 단어의 ascii 찾기
        while(final>122):
            final -= 26
        answer += chr(final)
    return answer