0과 1의 배열 num과 정수 k가 있다고 가정하고 모든 1이 서로 최소한 k 자리 떨어져 있는지 확인해야 합니다. 그렇지 않으면 False를 반환합니다.
따라서 입력이 nums =[1,0,0,0,1,0,0,1], k =2와 같으면 각각의 1이 에서 최소 2자리 떨어져 있으므로 출력은 참이 됩니다. 서로.
이 문제를 해결하기 위해 다음 단계를 따릅니다. −
-
마지막 :=-1
-
initialize i :=0의 경우, i <숫자의 크기일 때 업데이트(i를 1만큼 증가), 수행 -
-
nums[i]가 1과 같으면 -
-
마지막이 -1 또는 (i - last - 1)>=k와 같으면 -
-
마지막 :=나는
-
-
그렇지 않으면
-
거짓을 반환
-
-
-
-
true를 반환
예시
더 나은 이해를 위해 다음 구현을 살펴보겠습니다. −
#include <bits/stdc++.h> using namespace std; class Solution { public: bool kLengthApart(vector<int>& nums, int k) { int last = -1; for (int i = 0; i < nums.size(); i++) { if (nums[i] == 1) { if (last == -1 || (i - last - 1) >= k) last = i; else return false; } } return true; } }; main(){ Solution ob; vector<int> v = {1,0,0,0,1,0,0,1}; cout << (ob.kLengthApart(v, 2)); }
입력
{1,0,0,0,1,0,0,1}
출력
1