[프로그래머스] Lv.0 가까운 1 찾기.java

가까운 1 찾기.java

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

i가 idx보다 크거나 같고, arr의 i번째 원소가 1이라면 answer에 i를 반환합니다.

출처: 프로그래머스 코딩 테스트 연습, https://school.programmers.co.kr/learn/challenges