LeetCode 1437 - 是否所有 1 都至少相隔 k 个元素（周赛）

题目描述

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

解法：（Python）

把所有 1 的下标摘出来处理就行

class Solution:
    def kLengthApart(self, nums: List[int], k: int) -> bool:
        indx = [i for i, x in enumerate(nums) if x == 1]
        for i in range(len(indx)-1):
            if indx[i+1] - indx[i] - 1 >= k: continue
            else: return False
        return True