[프로그래머스][LV.1] 가장 가까운 같은 글자 | python3

2025. 1. 20. 15:05프로그래머스/LV.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을 완성해주세요.

제한사항
- 1 ≤ s의 길이 ≤ 10,000
  • s은 영어 소문자로만 이루어져 있습니다.

 


문제 풀이

같은 문자끼리의 인덱스 값이 차례대로 하나씩 필요하다. 딕셔너리를 만들어서 카값에 문자를 벨류값에 인덱스값을 넣는다. 새로운 문자면 키값-벨류값을 지정해주고 같은 문자면 벨류값을 업데이트 해준다.

answer 리스트에는 새로운 문자면 -1을 같은 문자면 (지금인덱스-지금을 제외한 가장 최신 인덱스)값을 넣어줘야 하기 때문에 for문과 딕셔너리를 활용하여 계산을 해주고 업데이트 한다.

 

내코드

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

 

 


더보기

다른사람풀이

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

    return answer

내 코드에서 dic[s[i]] = i 이부분이 if 와 else문에 중복으로 들어가 있다. 생략해서 if문 밖으로 하나만 써줘도 된다.