[프로그래머스/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은 영어 소문자로만 이루어져 있습니다.

 

입출력 예

 

 

나의 풀이1 (실패)

def solution(s):
    arr = [0] * 26
    answer = []
    
    for i in range(len(s)):
        j = ord(s[i]) - 97
        
        if arr[j]:
            answer.append(i - arr[j])
            arr[j] = i
        else:
            answer.append(-1)
            arr[j] = i
        
    return answer

• 테스트 케이스 30개 중에 3개만 통과했다.

 

나의 풀이2 (성공)

def solution(s):
    arr = [-1] * 26
    answer = []
    
    for i in range(len(s)):
        j = ord(s[i]) - 97
        
        if arr[j] == -1:
            answer.append(-1)
            arr[j] = i
        else:
            answer.append(i - arr[j])
            arr[j] = i
        
    return answer

알파벳의 개수만큼을 길이로 가지는 리스트를 만들어서

각 알파벳의 위치마다 가장 최근에 등장했던 인덱스의 위치를 저장했다.

`ord()`를 통해 해당 문자에 해당하는 유니코드를 알파벳의 인덱스를 구하는 데 이용한다.

 

만약 `arr`의 초기값을 `0`으로 설정하면

문자열의 첫 번째 글자의 인덱스를 저장하는 데 문제가 생긴다.

그래서 문자열의 인덱스의 영향을 받지 않도록 `arr`의 초기값을 `-1`로 변경했다.

 

다른 사람의 풀이

def solution(s):
    dict = {}
    answer = []
    
    for i in range(len(s)):
        if s[i] in dict:
            answer.append(i - dict[s[i]])
        else:
            answer.append(-1)
            
        dict[s[i]] = i
        
    return answer

내가 풀었던 것과 접근 방식은 비슷한데 리스트가 아닌 딕셔너리를 사용했다.