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

문제

문제

---

코드

#include <iostream>
#include <vector>
using namespace std;

int solution(vector<int> arr, int idx) {
    int answer = -1;

    if (arr[idx] == 1)
    {
        return idx;
    }
    else
    {
        for (int i = 0; i < arr.size(); ++i)
        {
            if (arr[i] == 1 && i > idx)
            {
                return i;
            }
        }
    }
    return answer;
}

---