프로그래머스 - java 가까운 1 찾기

💡 idx 보다 크면서 같은 인덱스값을 찾는건데

이 경우에는 조건의 시작을 idx 로 잡아주면 편하다!!

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;
    }
}

 

 

실패한 코드

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

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