1. 问题描述:
给你一个由若干 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
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/check-if-all-1s-are-at-least-length-k-places-away
2. 思路分析:
题目不难理解,最容易想到的是暴力破解,依次检查上一个1与当前1的距离是否满足要求,假如不满足直接return即可,所以需要使用一个临时的变量来存储上一个1出现的位置,这样才可以与当前位置的1进行距离的比较
3. 代码如下:
class Solution {
public boolean kLengthApart(int[] nums, int k) {
int pre = -1;
for (int i = 0; i < nums.length; ++i){
if (nums[i] == 1){
if (pre == -1) pre = i;
else if (i - pre - 1 < k) return false;
else pre = i;
}
}
return true;
}
}