(Python) 둘만의 암호 (프로그래머스 Lv.1)

둘만의 암호 (프로그래머스 Lv.1)

https://school.programmers.co.kr/learn/courses/30/lessons/155652

프로그래머스

 

내 코드

def change(i): #알파벳을 넘어가는 경우 확인
    if ord('z')<ord(i):
        return chr(ord(i)-26)
    else:
        return i

def add(i,skip,index): #index만큼 kip빼고 증가시켜주는 함수
    n = 0
    while(n<index):
        if change(chr(ord(i)+1)) in skip:
            i = change(chr(ord(i)+1))
        else:
            i = change(chr(ord(i)+1))
            n += 1
    return i

def solution(s, skip, index):
    answer = ''
    s = list(s)
    for i in range(len(s)):
        answer += add(s[i],skip,index)
    return answer

코드가 복잡해질것같아 두 함수를 따로 작성해서 분리시켜주었다.