[프로그래머스]가장 가까운 글자 - 파이썬

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

문제 설명

문자열 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은 영어 소문자로만 이루어져 있습니다.

---

입출력 예

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

---

입출력 예 설명

입출력 예 #1

지문과 같습니다.

입출력 예 #2

설명 생략

문제풀이

간단하게 생각해보면 s 의 각 요소를 도는 for 문을 만들고, 각 요소가 리스트 안에 담겨있지 않으면 answer 에 -1 을 추가하고, 담겨있다면 각 요소의 인덱스끼리 빼서 answer 에 추가하는 방법을 쓰면 될 거 같은데, 문자가 중복되는 경우 어떻게 가장 최근에 추가된 문자의 인덱스에 접근할 수 있을까?

• 딕셔너리를 사용해볼까?

1. result 값들을 담는 빈 리스트 answer, s 의 각 요소들을 담을 딕셔너리 arr
2. 각 요소에 대해 인덱스와 값에 동시에 접근하기 위해 enumerate(인덱스=i, 값=v) 함수를 사용.
3. 만약 arr 에 v 가 없다면? → arr[v] = i 추가, answer 에는 -1 추가하기
4. 있다면? → answer 에 i - arr[v] 추가하고, arr 에는 현재 v 의 인덱스와 값으로 업데이트!

최종코드

def solution(s):
    answer = []
    arr = {}
    
    for i, v in enumerate(s):
        if v not in arr:
            arr[v] = i
            answer.append(-1)
        else:
            answer.append(i - arr[v])
            arr[v] = i
    
    return answer