LeetCode contest 187 1437. 是否所有 1 都至少相隔 k 个元素 Check If All 1's Are at Least Length K Places Away

Table of Contents

一、中文版

二、英文版

三、My answer

四、解题报告

---

一、中文版

给你一个由若干 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

二、英文版

Given an array nums of 0s and 1s and an integer k, return True if all 1's are at least k places away from each other, otherwise return False.

 

Example 1:

 

Input: nums = [1,0,0,0,1,0,0,1], k = 2
Output: true
Explanation: Each of the 1s are at least 2 places away from each other.

Example 2:

Input: nums = [1,0,0,1,0,1], k = 2
Output: false
Explanation: The second 1 and third 1 are only one apart from each other.

Example 3:

Input: nums = [1,1,1,1,1], k = 0
Output: true

Example 4:

Input: nums = [0,1,0,1], k = 1
Output: true

 

Constraints:

• 1 <= nums.length <= 10^5
• 0 <= k <= nums.length
• nums[i] is 0 or 1

三、My answer

class Solution:
    def kLengthApart(self, nums: List[int], k: int) -> bool:
        
        index_list = []
        for i in range(len(nums)):
            if nums[i] == 1:
                index_list.append(i)

        for i in range(1,len(index_list)):
            subtract = index_list[i] - index_list[i-1]
            if subtract - 1 < k:
                return False
        return True

四、解题报告

用一个 index_list 来保存所有 1 出现的位置。

遍历 index_list，后项减前项做差，判断差值与 k 的关系即可。