题目描述

给你一个由若干 0 和 1 组成的数组 nums 以及整数 k。如果所有 1 都至少相隔 k 个元素，则返回 True ；否则，返回 False 。

样例

输入：nums = [1,0,0,0,1,0,0,1], k = 2
输出：true
解释：每个 1 都至少相隔 2 个元素。

输入：nums = [1,0,0,1,0,1], k = 2
输出：false
解释：第二个 1 和第三个 1 之间只隔了 1 个元素。

输入：nums = [1,1,1,1,1], k = 0
输出：true

输入：nums = [0,1,0,1], k = 1
输出：true

提示：

• 1 <= nums.length <= 10^5
• 0 <= k <= nums.length
• nums[i] 的值为 0 或 1`

算法分析

模拟

• last表示上一个1的位置，初始值为-1
• 枚举所有元素，若该位置i元素是1，判断i - last - 1 >= k是否满足(第一个1除外)，若满足继续判断下一个，不满足return false，更新last = i，所有都满足return true，

时间复杂度 $O(n)$

Java 代码

class Solution {
    public boolean kLengthApart(int[] nums, int k) {
        int n = nums.length;
        int last = - 1;
        for(int i = 0;i < n;i ++)
        {
            if(nums[i] == 1)
            {
                if(last != -1 && i - last - 1 < k) return false;
                last = i;
            }
        }
        return true;
    }
}