每日一题-是否所有 1 都至少相隔 k 个元素

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


题目

给你一个由若干 0 和 1 组成的数组 nums 以及整数 k。如果所有 1 都至少相隔 k 个元素,则返回 True ;否则,返回 False 。

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/check-if-all-1s-are-at-least-length-k-places-away
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

示例

示例 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 <= 105
0 <= k <= nums.length
nums[i] 的值为 0 或 1

解题

双指针的解法

class Solution:
    def kLengthApart(self, nums, k) -> bool:
        if 1 not in nums:
            return True
        left_index = nums.index(1)
        right_index = left_index + 1
        while right_index < len(nums):
            if nums[right_index] != 1:
                right_index += 1
            elif nums[right_index] == 1:
                if right_index - left_index <= k:
                    return False
                left_index = right_index
                right_index += 1
        return True

猜你喜欢

转载自blog.csdn.net/qq_33537936/article/details/113870977
今日推荐