给定一个由若干 0 和 1 组成的数组 A,我们最多可以将 K 个值从 0 变成 1 。
返回仅包含 1 的最长(连续)子数组的长度。
示例 1:
输入:A = [1,1,1,0,0,0,1,1,1,1,0], K = 2
输出:6
解释:
[1,1,1,0,0,1,1,1,1,1,1]
粗体数字从 0 翻转到 1,最长的子数组长度为 6。
示例 2:
输入:A = [0,0,1,1,0,0,1,1,1,0,1,1,0,0,0,1,1,1,1], K = 3
输出:10
解释:
[0,0,1,1,1,1,1,1,1,1,1,1,0,0,0,1,1,1,1]
粗体数字从 0 翻转到 1,最长的子数组长度为 10。
分析:
最长连续,想到滑动窗口,遇到1向右扩张,遇到0就要跟翻转次数K讨论,若K>0说明可以直接翻转这个0,若等于0说明没有翻转次数了,左指针必须移动到目前窗口中最左边的值为0的位置右侧(窗口收缩),然后翻转次数加1,窗口才能继续向右移动
PS:滑动窗口的right随着while一次次向右移动,核心在于如何确定左侧什么情况下移动,该移动多少。
public:
int longestOnes(vector<int>& A, int K) {
int maxOnes = 0;
// int tempOnes = 0;
int reverseTimes = K;
int left = 0;
int right = 0;
while(right < A.size()){
if(A[right] == 0){
// 没有翻转次数且又遇到了0
if(reverseTimes == 0){
// maxOnes = max(right - left, maxOnes);
if(A[left] == 0){
left++;
reverseTimes++;
}else{
while(A[left] == 1){
left++;
}
left++;
reverseTimes++;
}
}
reverseTimes--;
}
// 每次都比较一下
right++;
maxOnes = max(right - left, maxOnes);
// cout<< right - left << " ";
}
return maxOnes;
}