1004. 最大连续1的个数 III

题目描述:

1004. 最大连续1的个数 III_第1张图片

主要思路:

刚看到这个问题首先想到的是二分答案,二分长度,然后利用滑动窗口判断是否可以达成。

class Solution {
public:
    bool find(int x,vector<int> nums, int k)
    {
        int now=0;
        for(int i=0,j=0;i<nums.size();++i)
        {
            if(i-j>=x)
                if(nums[j++]==1)
                    now-=1;
            if(nums[i]==1)
                now+=1;
            if(now+k>=x)
                return true;
        }
        return false;
    }
    int longestOnes(vector<int>& nums, int k) {
        int l=0,r=nums.size();
        while(l<r)
        {
            int mid=(l+r+1)>>1;
            if(find(mid, nums, k))
                l=mid;
            else
                r=mid-1;
        }
        return l;
    }
};

然后看了题解,发现可以利用0的个数进行滑动窗口的限定。

class Solution {
public:
    int longestOnes(vector<int>& nums, int k) {
        int n=nums.size();
        int now=0,ans=0;
        for(int i=0,j=0;i<n;++i)
        {
            now += 1-nums[i];
            while(now>k)
                now-=(1-nums[j++]);
            ans=max(ans,i-j+1);
        }
        return ans;
    }
};

你可能感兴趣的:(Leetcode,算法,leetcode,职场和发展)