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

[문제 설명]

 

두 문자열 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):
    alpha = "abcdefghijklmnopqrstuvwxyz"
    
    u = []
    for ch in alpha:
        if ch not in skip:
            u.append(ch)
    #alpha 중 skip에 있는 알파벳을 제외한 전체 알파벳을 담는 배열 u
            
    answer = ""
    
    for ch in s: #주어진 문자열 s에서 하나씩 꺼내 index만큼 뒤에 있는 문자 추출
        current_index = u.index(ch) #현재 인덱스 = 현재 꺼낸 문자의 u에서의 인덱스
        new_index = (current_index+index)%len(u) #새 인덱스 = 현재 인덱스에서 뒤로 index만큼 미룸
        #u의 길이를 넘어가면 0부터 다시 시작
        answer += u[new_index] #뒤로 미룬 문자 answer에 추가
        
    return answer

먼저 문제를 봤을때,

- 문자열 s의 각 알파벳을 index만큼 뒤의 알파벳으로 바꿔줍니다.
- index 만큼의 뒤의 알파벳이 z를 넘어갈 경우 다시 a로 돌아갑니다.

- skip에 있는 알파벳은 제외하고 건너뜁니다.

 

 

[입출력 예]

s	skip	index	result
"aukks"	"wbqd"	5	"happy"

예시의 경우

u 배열에는 wbqd를 제외한 알파벳이 담긴다.

그리고 s에서 문자를 하나씩 꺼낸다.

첫번째 실행시

current_index = u.index('a') -- current_index = 0

new_index = (current_index + index) % len(u) -- new_index = 5

answer += u[5] -- answer ="h" skip문자인 b와 d를 제외하고 5칸 뒤에 있는 h

 

s에 있는 문자에 해당 로직을 적용한 후 answer에 담아 출력하면

result는 "happy"가 된다.

 

[시간복잡도]

def solution(s, skip, index):
    alpha = "abcdefghijklmnopqrstuvwxyz"
    
    u = []
    for ch in alpha:
        if ch not in skip:
            u.append(ch)
            
    #alpha의 길이(26) x skip의 길이(최대 10)번 반복
	# O(26x10) => O(1)
    
    answer = ""
    
    for ch in s: 
        current_index = u.index(ch) 
        new_index = (current_index+index)%len(u) 
        answer += u[new_index] 
        
   	#s의 길이(최대 50), u.index(ch)는 리스트를 처음부터 찾으므로 u의 길이만큼 탐색
    	#u의 길이는 최대 26
	#O(|s| x 26) => O(|s|)
        
    return answer

즉, 시간복잡도는 O(|s|)이다.

여기서는 s가 최대 50으로 정해져있어서 O(1)이라고 볼 수도 있지만

s의 길이가 바뀌면 문자열 길이에 비례해서 시간이 늘어나므로

O(|s|)라고 볼 수 있다. -- | |는 문자열의 길이를 의미하는 표기법이다.