给你一个由若干 0 和 1 组成的数组 nums 以及整数 k。如果所有 1 都至少相隔 k 个元素,则返回 True ;否则,返回 False 。
示例 1:
输入:nums = [1,0,0,0,1,0,0,1], k = 2
输出:true
解释:每个 1 都至少相隔 2 个元素。
示例 2:
输入:nums = [1,0,0,1,0,1], k = 2
输出:false
解释:第二个 1 和第三个 1 之间只隔了 1 个元素。
示例 3:
输入:nums = [1,1,1,1,1], k = 0
输出:true
示例 4:
输入:nums = [0,1,0,1], k = 1
输出:true
提示:
1 <= nums.length <= 10^5
0 <= k <= nums.length
nums[i] 的值为 0 或 1
class Solution {
public:
bool kLengthApart(vector<int>& nums, int k) {
int i, previous;
for (i = 0; i < nums.size() && nums[i] == 0; ++i)
;
if (i == nums.size())
return true;
previous = i;
for (i = i + 1; i < nums.size(); ++i) {
if (nums[i] == 1) {
if (i - previous - 1 >= k)
previous = i;
else
return false;
}
}
return true;
}
};