carloscn / structstudy

Leetcode daily trainning by using C/C++/RUST programming.
4 stars 1 forks source link

leetcode1437:是否所有 1 都至少相隔 k 个元素(check-if-all-1s-are-at-least-length-k-places-away) #228

Open carloscn opened 1 year ago

carloscn commented 1 year ago

问题描述

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

 

示例 1:

image

输入:nums = [1,0,0,0,1,0,0,1], k = 2 输出:true 解释:每个 1 都至少相隔 2 个元素。

示例 2:

image

输入: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/problems/check-if-all-1s-are-at-least-length-k-places-away 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

carloscn commented 1 year ago

问题分析

pub fn k_length_apart(nums: Vec<i32>, k: i32) -> bool
{
    if nums.len() < 1 {
        return false;
    }

    let mut count_0:i32 = k + 1;

    for i in 0..nums.len() {
        if nums[i] == 1 {
            if count_0 < k {
                return false;
            }
            count_0 = 0;
        } else {
            count_0 += 1;
        }
    }

    return true;
}
carloscn commented 1 year ago

code

https://review.gerrithub.io/c/carloscn/structstudy/+/554180 https://github.com/carloscn/structstudy/commit/07e8d4ab7200165609bfac6dbeccf34b8102afdd