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

最新推荐文章于 2021-05-01 13:29:10 发布

dos diosas

最新推荐文章于 2021-05-01 13:29:10 发布

阅读量197

点赞数

分类专栏： # LeetCode

本文链接：https://blog.csdn.net/dosdiosas_/article/details/106663931

版权

LeetCode 专栏收录该内容

273 篇文章 0 订阅

订阅专栏

class Solution {
public:
    bool kLengthApart(vector<int>& nums, int k) {
        vector<int>pos;
        for(int i=0;i<nums.size();i++)
        {
            if(nums[i]==1)
            {
                pos.push_back(i);
            }
        }
        if(pos.size()==0)
        {
            return true;
        }
        if(pos.size()==1)
        {
            return true;
        }
        for(int i=0;i<pos.size()-1;i++)
        {
            if(pos[i+1]-pos[i]<(k+1))
            {
                return false;
            }
        }
        return true;
    }
};