프로그래머스: 둘만의 암호

[level 1] 둘만의 암호 - 155652

문제 링크

성능 요약

메모리: 10.2 MB, 시간: 0.01 ms

구분

코딩테스트 연습 > 연습문제

채점결과

정확성: 100.0
합계: 100.0 / 100.0

제출 일자

2024년 1월 3일 6:8:2

문제 설명

두 문자열 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
본문 내용과 일치합니다.

풀이

def solution(s, skip, index):
    answer = ''
    alpha = 'abcdefghijklmnopqrstuvwxyz'
    idx = 0
    for skip_word in skip:
        alpha = alpha.replace(skip_word, '')

    for char in s:
        idx = (alpha.find(char) + index) % len(alpha)
        answer += alpha[idx]
    return answer

느낀점

처음 시도에는 얼추 비슷하다 생각해서 제출을 했는데 역시나 틀린게 많았다. 반례를 보며 생각을 해보니 예를 들어, 'y', 'za', 1이 주어졌을 때, 1을 더한 값은 'z'이고 'z'는 'za'에 하나만 존재하니 'a'를 출력하게 하는 잘못된 알고리즘이였다.

근데 누군가가 skip으로 주어진 단어들을 먼저 제거하고 생각해보라는 소리를 듣고, 바로 위와 같은 풀이가 떠올랐다.

조금 더 유연하게 생각하고 싶은데 말처럼 쉽지 않은것 같아서 아쉽다..