[프로그래머스] Lv.1 가장 가까운 같은 글자 [Python] - 문자열

주어진 문자열의 각 글자가 처음 나오면 -1,

이미 나온 글자는 바로 앞의 글자와의 거리를 리스트에 담아서 반환하는 문제

 

문제 설명
문자열 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을 완성해주세요.

제한사항
     1 ≤ s의 길이 ≤ 10,000
          s은 영어 소문자로만 이루어져 있습니다.

https://school.programmers.co.kr/learn/courses/30/lessons/142086

 

거리를 구하기 위해서는 발견한 글자의 현재 인덱스와 이전 인덱스의 차를 구하면 됩니다.

이전 인덱스를 기억하기 위해서 딕셔너리를 사용해서 코드를 작성했습니다.

def solution(s):
    d = {}	# 글자를 key로 현재 발견된 글자의 인덱스를 value로 저장할 딕셔너리
    li = []	# 정답을 담을 리스트
    for i, k in enumerate(s):
        if k not in d:	# 처음 보는 글자는 -1
            li.append(-1)
        else:		# 기존에 발견했던 글자는 현재 익덱스 - 이전 최근 인덱스 = 떨어진 거리
            li.append(i-d[k])
        d[k] = i	# 글자의 인덱스 저장
    return li

 

입출력 예

s	result
"banana"	[-1, -1, -1, 2, 2, 2]
"foobar"	[-1, -1, 1, -1, -1, -1]