프로그래머스 - '둘만의 암호' [코딩테스트 연습]

# 둘만의 암호 # 155652 # lv1 # Python

 

# 문제

---

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

 

# 풀이

---

# 첫 번째 풀이 리스트 활용 풀이
def solution(s, skip, index):
	""" 아이디어 """
    # 0. 인덱스를 통한 요소 접근을 편하게 하고자 문자열을 리스트로 변환.
    # 1. skip에 있는 문자들을 alpha_list에서 삭제.
    # 2. 문자열 s의 각 알파벳을 index 만큼의 뒤의 알파벳으로 바꿔줌.
    # 3. 만약 새로 바뀐 리스트의 인덱스값이 alpha_list의 길이보다 길면, 긴만큼을 빼줌.
    # 단, index 의 최대 길이는 20, alpha_list의 길이는 26이므로, 이를 예외 처리를 통해 확인해야 함. 
    
    alpha_list = list('abcdefghijklmnopqrstuvwxyz')
    s_list = list(s)
    answer = ''
    for i in skip:
        alpha_list.remove(i)
    
    for i in range(len(s_list)):
        new_idx = alpha_list.index(s_list[i]) + index        
        while new_idx >= len(alpha_list):
            new_idx -= len(alpha_list)
        s_list[i] = alpha_list[new_idx]
    
    return answer.join(s_list)

# 두 번째 풀이. 리스트 x, 문자열 직접 접근
def solution(s, skip, index):
    """ 아이디어는 동일 """
    
    alpha_ = 'abcdefghijklmnopqrstuvwxyz'
    answer = ''
    for i in skip:
        alpha_ = alpha_.replace(i, '')
    
    for i in s:
        new_idx = alpha_.find(i) + index        
        while new_idx >= len(alpha_) - 1:
            new_idx -= len(alpha_)
        answer += alpha_[(new_idx)]

    return answer