문자열 s가 주어졌을 때, s의 각 위치마다 자신보다 앞에 나왔으면서, 자신과 가장 가까운 곳에 있는 같은 글자가 어디 있는지 반환하는 함수 구현하기
문자열 s
가 주어졌을 때, s
의 각 위치마다 자신보다 앞에 나왔으면서, 자신과 가장 가까운 곳에 있는 같은 글자가 어디 있는지 알고 싶습니다.
예를 들어, s
="banana"라고 할 때, 각 글자들을 왼쪽부터 오른쪽으로 읽어 나가면서 다음과 같이 진행할 수 있습니다.
문자열 s
이 주어질 때, 위와 같이 정의된 연산을 수행하는 함수 solution을 완성해주세요.
s
의 길이 ≤ 10,000
s
은 영어 소문자로만 이루어져 있습니다.
s | result |
"banana" | [-1, -1, -1, 2, 2, 2] |
"foobar" | [-1, -1, 1, -1, -1, -1] |
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