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

입출력 예시

s result
"banana" [-1, -1, -1, 2, 2, 2]
"foobar" [-1, -1, 1, -1, -1, -1]

풀이

1. s에서 추출된 문자가 앞에 다른 문자들이 있으면 비교해가면서 -1 혹은 가장 가까운 동일한 문자와의 index차이를 구해주면 된다.

def solution(s):
    answer = []
    
    for i in range(len(s)):
        answer.append(0)

    for i1, s1 in enumerate(s):
        temp = []
        for i2, s2 in enumerate(s):
            if s1 == s2:
                temp.append(i2)
                
        answer[temp[0]] = -1
        for i3, num in enumerate(temp):
            if answer[num] == 0:
                answer[num] = temp[i3]-temp[i3-1]              
                
    return answer

• answer 리스트에 s의 길이만큼 0으로 초기화해준다.(나중에 요소들을 index로 추가하기 위함)
• for문으로 s를 enumerate()함수로 인덱스와 함께 추출해주고, temp 리스트 변수를 초기화해준다. 각 문자마다 temp가 초기화되어 사용해줄 것이다.
• 두 번째 for문도 s를 사용해주는데 이 때 추출된 s1과 s2가 같으면 temp리스트에 두 번째에 추출한 index i2를 추가해준다.
• temp에는 동일한 문자들의 index가 저장되어있는데 이 때 answer의 temp의 첫 번째 인자인 [0]에는 -1을 저장해준다.(맨 앞 문자에는 -1을 리턴하기 위함)
• 다음 temp를 enumerate()함수로 추출해 answer의 temp내 index에 값을 넣어주게 되는데 이때 만약 answer에 값이 있으면 적용되지 않게 if문을 사용해준다.
• answer의 값은 해당문자와 그 앞문자의 index 차이값이 들어가야하므로 temp[해당문자 index]-temp[해당문자 index-1]을 해주면 그 차이값을 구할 수 있다.

오답노트

• 해당 문자와 그 앞에 문자가 있는지 확인을 먼저 해주어야했고
• 반복문이 필요 이상으로 많이 돌아가게되는데 그건 좀 해결해주면 좋을거 같다.