[python_study] 가장 가까운 같은 글자

인증사진(2024-07-28)

문제

문자열 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 = [-1] * len(s)
    char_indices = {}
    
    for i, char in enumerate(s):
        if char in char_indices:
            closest_index = float('inf')
            for index in char_indices[char]:
                closest_index = min(closest_index, i - index)
            answer[i] = closest_index
        if char not in char_indices:
            char_indices[char] = []
        char_indices[char].append(i)
    
    return answer

 

추가공부

answer 리스트 초기화 ->

char_indices : 각 문자가 등장한 인덱스들의 리스트 저장

enumerate(s)를 : 문자열 s를 순회하면서 각 문자의 인덱스(i)와 문자(char) 가져오기

현재 문자 char가 char_indices에 이미 존재하는 경우와 그렇지 않은 경우로 나누고,

문자열 s를 왼쪽에서 오른쪽으로 순회하면서 각 문자의 인덱스를 확인.

이전에 등장한 같은 문자들의 가장 가까운 위치를 answer 리스트에 업데이트함.