Programmers

19. 가장 가까운 같은 글자

커다란숲 2023. 6. 21. 16:41
728x90

문제 설명

문자열 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):
    answer = []
    s = list(s)
    dic_ = {}
    for i in range(len(s)):
        if (i == 0) or s[i] not in s[:i]:
            answer.append(-1)
        else:
            answer.append(i - dic_[s[i]])
        dic_[s[i]] = i
    return answer

처음에는 index 메서드를 사용해서 구현하려고 했는데, 테스트 결과 index는 해당 리스트에서 그 단어가 가장 처음 나온 위치를 반환하기 때문에 적합하지 않았다. 그래서 딕셔너리를 생성하여, 앞 단어의 위치를 갱신하게 만들었다. 

728x90