[프로그래머스] 가까운 1 찾기 - 파이썬

Programmers

가까운 1 찾기

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

프로그래머스

문제

코드

<풀이1>

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

 

<풀이2>

• 풀이1의 조금 더 정리된 코드

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

Reference

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