[프로그래머스] Lv.1 가장 가까운 같은 글자 (파이썬)

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

---

나의 풀이

def solution(s):
    alp = []
    answer = []
    for i in list(s):
        if i not in alp:
            alp.append(i)
            answer.append(-1)
        else:
            index = [i for i, x in enumerate(list(s)) if x == i]
            z = [s.index(i)-k for k in index]
            ans = min([x for x in z if x!=0])
            answer.append(ans)
    return answer

 

문제 1. z가 비어있을 때 min 함수가 작동이 안 됨

문제 2. z 선언 부분에서 s.index(i)는 가장 앞에 있는 값만 찾아서 안 됨

 

 

다른 사람의 풀이

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

 

# location[char]로 업데이트 하는 부분 기억할 것 !

# enumerate 적극 활용할 것 ! (똑똑하게...)