comments | difficulty | edit_url | rating | source | tags | |
---|---|---|---|---|---|---|
true |
简单 |
1193 |
第 187 场周赛 Q2 |
|
给你一个由若干 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 个元素。
提示:
1 <= nums.length <= 105
0 <= k <= nums.length
nums[i]
的值为0
或1
我们可以遍历数组 false
;否则,将
遍历结束后,返回 true
。
时间复杂度
class Solution:
def kLengthApart(self, nums: List[int], k: int) -> bool:
j = -inf
for i, x in enumerate(nums):
if x:
if i - j - 1 < k:
return False
j = i
return True
class Solution {
public boolean kLengthApart(int[] nums, int k) {
int j = -(k + 1);
for (int i = 0; i < nums.length; ++i) {
if (nums[i] == 1) {
if (i - j - 1 < k) {
return false;
}
j = i;
}
}
return true;
}
}
class Solution {
public:
bool kLengthApart(vector<int>& nums, int k) {
int j = -(k + 1);
for (int i = 0; i < nums.size(); ++i) {
if (nums[i] == 1) {
if (i - j - 1 < k) {
return false;
}
j = i;
}
}
return true;
}
};
func kLengthApart(nums []int, k int) bool {
j := -(k + 1)
for i, x := range nums {
if x == 1 {
if i-j-1 < k {
return false
}
j = i
}
}
return true
}
function kLengthApart(nums: number[], k: number): boolean {
let j = -(k + 1);
for (let i = 0; i < nums.length; ++i) {
if (nums[i] === 1) {
if (i - j - 1 < k) {
return false;
}
j = i;
}
}
return true;
}