[Python] 프로그래머스 가장 가까운 같은 글자

문제 설명

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

 

https://school.programmers.co.kr/learn/courses/30/lessons/142086?language=python3

프로그래머스

제한 사항

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

예제

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

문제 풀이

def solution(s):
    answer = []
    D = {elem: [] for elem in set(s)}
    for i in range(len(s)):
        if not D[s[i]]:
            answer.append(-1)
            D[s[i]].append(i)
        else:
            answer.append(i - D[s[i]][-1])
            D[s[i]].append(i)
    return answer

딕셔너리에 각 문자의 인덱스를 저장하고,

현재 인덱스와 해당 문자의 마지막 인덱스의 차이를 계산하여 각 문자의 거리를 측정합니다.

def solution(s):
    answer = []
    D = {elem: -1 for elem in set(s)}
    
    for i, char in enumerate(s):
        if D[char] == -1:
            answer.append(-1)
        else:
            answer.append(i - D[char])
        D[char] = i
        
    return answer

---

파이썬을 독학하시는 분들에게 도움이 되길 바라며,
혹 더 좋은 방법이 있거나 오류가 있다면 편하게 말씀 부탁드립니다.