두 문자열 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 함수를 완성해주세요.
s | skip | index | result |
---|---|---|---|
"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 처리를 한번 더 했음에도 불구하고 시간초과가 따로 안나온걸 보면 이게 더 효율이 좋다!
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 하는 방식
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 코드 쓴 거 하나 더!
알아둬야하긴 하는데 맨날 까먹어서 큰일이애오