[프로그래머스] 가까운 1 찾기

 

<내가 쓴 코드>

def solution(arr, idx):
    answer = -1
    for i in range(len(arr)):
        if i >= idx and arr[i] == 1:
            answer = i
            break
    return answer

 

break 해야 한다!!! 이후에 1이 더 있으면 그 값이 출력되니까!!