Algorithm/알고리즘 문제

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

은 딩 2023. 10. 10. 00:17

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

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

 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr

 

 


아이디어

skip 할 문자들을 먼저 제거한 후

s를 for문으로 돌면서 주어진 index만큼 더해주고, skip 문자들 제거하고 남은 알파벳의 개수로 나눈 나머지를 구하면 된다!

 

z 다음 a로 넘어가야하는데 이걸 어떻게 하지?라는 생각이 들었지만 % 모듈러 이용하면 된다~!

 

 

 


코드

def solution(s, skip, index):
    answer = ''
    alpha = "abcdefghijklmnopqrstuvwxyz"
    
    # alpha 문자열에서 skip 문자들 제거
    for c in skip:
        alpha = alpha.replace(c, "")
    
    # 바뀔 문자 찾기
    for i in s:
        change = alpha[(alpha.index(i) + index) % len(alpha)]
        answer += change
        
    return answer