문제 설명

문자열 s가 주어졌을 때, s의 각 위치마다 자신보다 앞에 나왔으면서, 자신과 가장 가까운 곳에 있는 같은 글자가 어디 있는지 알고 싶습니다.

예를 들어, s="banana"라고 할 때,  각 글자들을 왼쪽부터 오른쪽으로 읽어 나가면서 다음과 같이 진행할 수 있습니다.

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

문자열 s이 주어질 때, 위와 같이 정의된 연산을 수행하는 함수 solution을 완성해주세요.

제한 사항

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

풀이

Python

def solution(s):
    # 각 문자의 마지막 위치를 저장하기 위한 dictionary 선언
    last_seen = {}
    # 결과를 저장할 리스트, 초기 값은 모두 -1
    answer = [-1] * len(s)
    
    # 문자열 s를 인덱스와 함께 반복
    for i, char in enumerate(s):
        # 현재 문자가 이전에 등장한 적이 있다면
        if char in last_seen:
            # 이전 위치와 현재 위치 사이의 거리 계산
            answer[i] = i - last_seen[char]
        # 현재 문자의 위치를 업데이트
        last_seen[char] = i
    
    # 계산된 결과 반환
    return answer