leetcode_1437 是否所有 1 都至少相隔 k 个元素

129 阅读1分钟

要求

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

示例 1:

image.png

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

示例 2:

image.png

输入: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:
    def kLengthApart(self, nums: List[int], k: int) -> bool:
        pre = None
        for i,num in enumerate(nums):
            if num == 1:
                if pre != None:
                    if i - pre - 1 < k:
                        return False
                pre = i
        return True

image.png

解题思路:扫描数组,用一个变量记录上一个1的下标,即可知道当前1和上一个1之间的元素个数。时间复杂度:O(N)空间复杂度:O(1)