[프로그래머스] level1 : 가장 가까운 같은 글자

파이썬이랑 친해지기

파이썬에서는 iteratable한 데이터에 대해 인덱스와 값을 함께 조회할 수 있는 enumerate() 라는 함수를 제공한다.

다음과 같이 작성할 수 있다.

for idx, ch in enumerate("hello world"):
    print(idx, ch)

# 실행결과    
# 0 h
# 1 e
# 2 l
# 3 l
# 4 o
# 5
# 6 w
# 7 o
# 8 r
# 9 l
# 10 d

 

코드

def solution(s):
    tmp = {}
    answer = []
    
    for idx, ch in enumerate(s):
        if tmp.get(ch) is None:
            tmp[ch] = idx
            answer.append(-1)
        else:
            answer.append(idx - tmp[ch])
            tmp[ch] = idx
    
    return answer