995. K 连续位的最小翻转次数
在仅包含 0 和 1 的数组 A 中,一次 K 位翻转包括选择一个长度为 K 的(连续)子数组,同时将子数组中的每个 0 更改为 1,而每个
1 更改为 0。返回所需的 K 位翻转的最小次数,以便数组没有值为 0 的元素。如果不可能,返回 -1。
示例 1:
输入:A = [0,1,0], K = 1
输出:2
解释:先翻转 A[0],然后翻转 A[2]。
示例 2:
输入:A = [1,1,0], K = 2
输出:-1
解释:无论我们怎样翻转大小为 2 的子数组,我们都不能使数组变为 [1,1,1]。
示例 3:
输入:A = [0,0,0,1,0,1,1,0], K = 3
输出:3
解释:
翻转 A[0],A[1],A[2]: A变成 [1,1,1,1,0,1,1,0]
翻转 A[4],A[5],A[6]: A变成 [1,1,1,1,1,0,0,0]
翻转 A[5],A[6],A[7]: A变成 [1,1,1,1,1,1,1,1]
解:
-
我们使用队列模拟滑动窗口,该滑动窗口的含义是前面
K - 1
个元素中,以哪些位置起始的 子区间 进行了翻转。
该滑动窗口从左向右滑动,如果当前位置 ii需要翻转,则把该位置存储到队列中。遍历到新位置 j (j < i + K)
时,队列中元素的个数代表了 i
被前面 K−1
个元素翻转的次数。
-
当
A[i]
为0
,如果i
位置被翻转了偶数次,那么翻转后仍是0
,当前元素需要翻转; -
当
A[i]
为1
,如果i
位置被翻转了奇数次,那么翻转后变成0
,当前元素需要翻转
综合上面两点,我们得到一个结论,如果 len(que) % 2 == A[i] 时,当前元素需要翻转。
当i+K>N
时,说明需要翻转大小为 K
的子区间,但是后面剩余的元素不到 K
个了,所以返回 -1
。
class Solution {
public:
int minKBitFlips(vector<int>& A, int K) {
int len=A.size();
queue<int> q;
int res=0;
for(int i=0;i<len;i++){
if(!q.empty() && i >= q.front()+K){
q.pop();
}
if(q.size()%2==A[i]){ //需要进行反转
if(i+K>len) return -1;
q.push(i);
res++; //次数++
}
}
return res;
}
};