가장 가까운 글자 - 런타임 에러

문제 설명

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

 

풀이

def solution(s):

    arr = [-1]
    for i in range(1,len(s)):
        cnt =0
        arr_num = []
        for j in range(0,i):
            if s[i] == s[j]:
                arr_num.append(i-cnt)
            elif len(arr_num) == 0:
                arr_num.append(-1)
            cnt += 1
        if arr_num.count(-1) != len(arr_num):
            arr_num.remove(-1)
        arr.append(arr_num[-1])

    answer = arr
    return answer

count()를 세면서 몇 번 앞쪽 숫자인지 세며, 노가다로 풀었더니 런타임 에러가 떴다.

def solution(s):
    arr = [-1]
    arr_num=[]
    for i in range(1,len(s)):
        for j in range(i,0,-1):
            if s[i] == s[j-1]:
                arr.append(i-(j-1))
                arr_num.append(i-(j-1))
                break
        if len(arr_num) == 0:
            arr.append(-1)
        arr_num =[]
            
    answer = arr
    return answer

count를 세지 않고 range( , ,-1) 로 거꾸로 문자를 조회하여 글자가 있으면 인덱스 번호 차이로 구하고, 없으면 -1을 반환했따.

없는 것을 구하기 위해 최종 결과인 arr 리스트와 별개로  arr_num으로 구해서 len(arr_num) == 0 으로 구했다.

통과는 됐지만 더 간단한 방법이 있을 것 같다.

def solution(s):
    answer = []
    t = list(s)
    for i in range(len(t)):
        if t[i] in t[:i]:
            answer.append(i-t.index(t[i]))
            t[t.index(t[i])] = '0'      #같은 글자의 첫 번째 글자를 없애기 위해
        else:
            answer.append(-1)
    return answer

처럼 for를 한 번만 써서 간단한 방법으로 조회할 수도 있다.

하지만 t [ t.index( t [ i ] ) ] 처럼 중복되는 첫 번째 숫자를 없애줘야하는 번거로움이 있다.

 

딕셔너리로 하면 중복 조회에 더 효과적이다. 

예를 들어

dic = {}
s = 'banana'
for i in range(len(s)):
    dic[s[i]] = i
    print(dic)

#  {'b': 0}
#  {'b': 0, 'a': 1}
#  {'b': 0, 'a': 1, 'n': 2}
#  {'b': 0, 'a': 3, 'n': 2}
#  {'b': 0, 'a': 3, 'n': 4}
#  {'b': 0, 'a': 5, 'n': 4}

중복되는 문자는 인덱스 번호만 갱신된다.

그러므로 앞서  딕셔너리에 등록된 문자가 있다면 

def solution(s):
    answer = []
    dic = {}               # 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

 

 

 

출처

https://school.programmers.co.kr/learn/courses/30/lessons/142086

프로그래머스