Lv1. 둘만의 암호

zz·2023년 5월 10일
0

프로그래머스

목록 보기
32/36

문제 설명

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

입출력 예

sskipindexresult
"aukks""wbqd"5"happy"

내 풀이

def solution(s, skip, index):
    answer = ''
    alphabet = 'abcdefghijklmnopqrstuvwxyz'
    # 전처리 - skip letters remove
    for char in skip:
        alphabet = alphabet.replace(char, "")
    # 이제 계산
    for letter in s: 
        answer += alphabet[(alphabet.index(letter) + index)%(26-len(skip))]
    return answer

원래는 이렇게 안하고 약간 다른방식으로 (많이는 안다르지만) 했었는데 시간 초과 나길래 음 알파벳 길이를 줄여봐야겠다해서 반으로 줄였더니 시간 내에 다 통과되었다

### 시간초과 코드!
def solution(s, skip, index):
    answer = ''
    alphabet = 'abcdefghijklmnopqrstuvwxyz'*2 # 2배 길이의 string
    # 전처리 - skip letters remove
    for char in skip:
        alphabet = alphabet.replace(char, "")
    # 이제 계산
    for letter in s: 
        answer += alphabet[(alphabet.index(letter) + index)] # 나눗셈을 안하고 바로 계산한 next char
    return answer

코드 중 다른부분은 주석이 달려있는 두 줄이다
2배 길이로 늘이다보니 메모리가 늘어나면서 자연스럽게 연산량이 늘어난 것 같다
string을 반으로 줄이고 index 처리를 한번 더 했음에도 불구하고 시간초과가 따로 안나온걸 보면 이게 더 효율이 좋다!

다른사람 코드

다른 코드 1

from string import ascii_lowercase

def solution(s, skip, index):
    result = ''

    a_to_z = set(ascii_lowercase)
    a_to_z -= set(skip)
    a_to_z = sorted(a_to_z)
    l = len(a_to_z)

    dic_alpha = {alpha:idx for idx, alpha in enumerate(a_to_z)}

    for i in s:
        result += a_to_z[(dic_alpha[i] + index) % l]

    return result

set을 이용해서 sorting 하는 방식

다른코드 2

def solution(s, skip, index):
    answer = ''
    arr = [chr(i) for i in range(97, 123) if chr(i) not in skip] * 10
    # print(arr, len(arr))
    for i in s:
        answer += arr[arr.index(i) + index]
    return answer

ascii 코드 쓴 거 하나 더!
알아둬야하긴 하는데 맨날 까먹어서 큰일이애오

profile
응애 나 애기개발자

0개의 댓글