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

문제

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

 

내가 생각한 풀이 방법

알파벳 소문자 배열을 만든다

알파벳 소문자 배열에서 skip을 뺀다 (for문 돌면서 not in)

이 부분 시간 복잡도가 엄청날 것 같은데,,,

 

s를 for문을 돌면서

알파벳 배열에서 i를 찾고 i+index 에 있는 수를 answer에 append한다

 

코드로 구현

1차 시도

def solution(s, skip, index):
    answer = ''
    alp_low = ['a','b','c','d','e','f','g','h','i','j','k','l','m','n','o','p','q','r','s','t','u','v','w','x','y','z']
    alp_lst = []
    for i in alp_low:
        if i not in skip:
            alp_lst.append(i)
    
    for i in s:
        i_idx = alp_lst.index(i)
        if i_idx + index > len(alp_lst) - 1:
            answer += alp_lst[(i_idx + index) - len(alp_lst)]
        else:
            answer += alp_lst[i_idx + index]
    return answer

- 런타임 에러...

- 78.9점

- alp_lst를 따로 만들지 말아야하나..?ㅠㅠ

 

2차 시도

def solution(s, skip, index):
    answer = ''
    alp_low = ['a','b','c','d','e','f','g','h','i','j','k','l','m','n','o','p','q','r','s','t','u','v','w','x','y','z']
    alp_lst = []
    for i in alp_low:
        if i not in skip:
            alp_lst.append(i)
    
    for i in s:
        i_idx = alp_lst.index(i)
        answer += alp_lst[(i_idx + index) % len(alp_lst)]
    return answer

- if else문 없애고 %로 계산했더니 성공

- 이게 이렇게 큰 차이가 날줄은 몰랐다..! 모듈로 연산 기억하기

 

 

다른 사람 풀이