Python/Coding Test

[Coding Test] 가장 가까운 같은 글자

gangee 2023. 8. 6. 16:36
728x90
반응형

프로그래머스 코딩테스트 1단계 : 가장 가까운 같은 글자

문제 설명

문자열 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을 완성해주세요.

문제 풀이

  • 글자의 인덱스를 확인하기위해 dict()로 빈 딕셔너리 생성
  • s의 길이 만큼 반복문 실행
  • 조건문을 통해 s의 글자가 딕셔너리에 있을 경우, i에서 s[i]의 a 딕셔너리 value 값을 빼고 answer에 추가
  • 없을 경우, -1을 answer에 추가
  • 조건문이 끝나고 나면 a 딕셔너리 value 값을 i로 변경하여 계속 반복
  • 최종적으로 나온 answer 값 반환

정답 코드

def solution(s):
    answer = []
    a = dict()

    for i in range(len(s)):
        if s[i] in a:
            answer.append(i - a[s[i]])
        else:
            answer.append(-1)
        a[s[i]] = i

    return answer
* 이 문제는 프로그래머스 코딩테스트 연습 1단계 문제입니다.
728x90
반응형