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