LeetCode：5401. 是否所有 1 都至少相隔 k 个元素

https://leetcode-cn.com/problems/check-if-all-1s-are-at-least-length-k-places-away

给你一个由若干 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
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

思路：首先记录1的位置，然后寻找相邻位置的最小距离，判断最小距离与k的关系

class Solution(object):
    def kLengthApart(self, nums, k):
        """
        :type nums: List[int]
        :type k: int
        :rtype: bool
        """
        position = []  # 1的位置
        for i, num in enumerate(nums):
            if num == 1:
                position.append(i)
        # 寻找两个1最小距离
        min_distance = 10**5+1
        for i, p in enumerate(position[:-1]):
            distance = position[i+1] - p - 1
            if distance < min_distance:
                min_distance = distance
        # 判断最小距离与k的关系
        if min_distance >= k:
            return True
        else:
            return False