Programmers/프로그래머스-가까운 1 찾기-Python/Go/Java

문제

Python Code

from typing import List

def solution(arr: List[int], idx: int) -> int:
    for x in range(idx, len(arr)):
        if arr[x] == 1:
            return x
    
    return -1

Go Code

func solution(arr []int, idx int) int {
    for i := idx; i < len(arr); i++ {
        if arr[i] == 1 {
            return i
        }
    }
    return -1
}

Java Code

class Solution {
    public int solution(int[] arr, int idx) {
        for (int i = idx; i < arr.length; i++) {
            if (arr[i] == 1) {
                return i;
            }
        }
        return -1;
    }
}

결과

문제출처 & 깃허브

programmers
Github