정수 배열 arr가 주어집니다. 이때 arr의 원소는 1 또는 0입니다. 정수 idx가 주어졌을 때, idx보다 크면서 배열의 값이 1인 가장 작은 인덱스를 찾아서 반환하는 solution 함수를 완성해 주세요.
단, 만약 그러한 인덱스가 없다면 -1을 반환합니다.
function solution(arr, idx) {
for (let i = idx; i < arr.length; i++) {
if (arr[i] === 1) return i;
}
return -1;
}
정석적으로 문제를 풀었다. i를 idx부터 시작하는 반복문을 통해 값이 1인 인덱스를 찾고, 찾지 못할 경우 -1을 return했다.
const solution=(a,i)=>a.indexOf(1,i);
function solution(arr, idx) {
return arr.findIndex((v, i) => idx <= i && v === 1);
}