题目描述
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: continueelse: return Falsereturn True