문제

문자열 s가 주어졌을 때, s의 각 위치마다 자신보다 앞에 나왔으면서, 자신과 가장 가까운 곳에 있는 같은 글자가 어디 있는지 알고 싶습니다.예를 들어, s="banana"라고 할 때, 각 글자들을 왼쪽부터 오른쪽으로 읽어 나가면서 다음과 같이 진행할 수 있습니다.

b는 처음 나왔기 때문에 자신의 앞에 같은 글자가 없습니다. 이는 -1로 표현합니다.
a는 처음 나왔기 때문에 자신의 앞에 같은 글자가 없습니다. 이는 -1로 표현합니다.
n은 처음 나왔기 때문에 자신의 앞에 같은 글자가 없습니다. 이는 -1로 표현합니다.
a는 자신보다 두 칸 앞에 a가 있습니다. 이는 2로 표현합니다.
n도 자신보다 두 칸 앞에 n이 있습니다. 이는 2로 표현합니다.
a는 자신보다 두 칸, 네 칸 앞에 a가 있습니다. 이 중 가까운 것은 두 칸 앞이고, 이는 2로 표현합니다.

따라서 최종 결과물은 [-1, -1, -1, 2, 2, 2]가 됩니다.
문자열 s이 주어질 때, 위와 같이 정의된 연산을 수행하는 함수 solution을 완성해주세요.

제출답안

def solution(s):
    answer = []
    dict = {} # 문자별로 리스트 형태의 인덱스 모음을 저장할 딕셔너리
    for i, char in enumerate(s): #인덱스번호, #문자
        try: # 딕셔너리에 등록되있는 경우
            dict[char].append(i)
            answer.append(i-dict[char][-2])
        except: # 딕셔너리에 등록되있지 않을 경우
            answer.append(-1) #answer에 -1
            dict[char] = [i] 
    
    return answer

제출 후 개선답안

각 문자별 인덱스 모두를 저장할 필요가 없음 >> 이전 인덱스를 무시하고 최신 인덱스만 저장
try, except 문을 if, else 문으로 변경

def solution(s):
    answer = []
    dict = {} # 문자별로 리스트 형태의 인덱스 모음을 저장할 딕셔너리
    for i, char in enumerate(s): #인덱스번호, #문자
        if s[i] not in dict:
            answer.append(-1)
        else:
            answer.append(i-dict[s[i]])
        dict[s[i]] = i
    
    return answer

문제 바로가기

728x90