[프로그래머스 lv1] 둘만의 암호

문제 보기

https://school.programmers.co.kr/learn/courses/30/lessons/155652?language=python3 

프로그래머스

 

문제 해결

- 처음 풀이 (오답)

def solution(s, skip, index):
    answer = ''
    tmp_word = 'abcdefghijklmnopqrstuvwxyz'
    
    for i in skip:
        tmp_word = tmp_word.replace(i, "")
    
    word = tmp_word * 2
    
    tmp = ''
    for i in s:
        tmp += word[word.index(i) + index]
   
    return tmp

replace() 함수를 사용하고 z를 넘어가면 a로 가야하므로 %len(word)를 해준다.

 

 

 

소스 코드 (정답 풀이)

def solution(s, skip, index):
    answer = ''
    word = 'abcdefghijklmnopqrstuvwxyz'
    
    for i in skip:
        word = word.replace(i, "")

    tmp = ''
    for i in s:
        idx = (word.index(i) + index) % len(word)
        tmp += word[idx]
   
    return tmp