[파이썬] 프로그래머스 마법의 엘리베이터

테스트 케이스는 다 맞았고 5개 정도 추가해서도 맞았는데 확인해보니 3개정도가 계속 틀렸다. a[i] 가 5인 경우 다음 숫자까지 고려해주었어야 했는데 그러지 못했다.

 

def solution(storey):
    answer = 0
 
    a = list(str(storey))
 
 
 
    for i in range(len(a)-1,-1,-1):
 
 
        if int(a[i])>5:
            answer += (10-int(a[i]))
 
            k = int(a[i-1])+1
            a[i-1] = k
 
 
 
 
        else:
 
            answer += int(a[i])
 
 
 
 
    check = ['1','2','3','4','5','6','7','8','9','0']
 
 
    if int(a[0])-1>=6 and a[0] not in check:
            answer+=1
 
 
 
 
 
 
    return answer