题目描述:
主要思路:
刚看到这个问题首先想到的是二分答案,二分长度,然后利用滑动窗口判断是否可以达成。
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;
}
};