Description
Solution
观察发现,当从头开始处理到第
i
i
i 位(前面的位数都为
1
1
1)时, 若该位(在考虑被之前的翻转所影响的情况下)为
0
0
0,则必须翻转,且是以从第
i
i
i 位开始向后K个长度的翻转(因为前面的位数都为
1
1
1了,再翻转它们没有任何意义),所以我们可以贪心的进行翻转操作,即可满足最少次数的要求。
实现上,采用双端队列的方式,动态维护能影响到当前位的所有翻转,根据现存翻转个数来判断该位的当前值(奇数个则取反,偶数个不变)
时间复杂度
O
(
N
)
O(N)
O(N)
空间复杂度
O
(
N
)
O(N)
O(N)
Code
class Solution {
public:
static const int maxn = 3e4 + 7;
int minKBitFlips(vector<int>& A, int K) {
int head = 0, tail = 0,que[maxn];
int n = A.size(), res = 0;
for(int i = 0;i < n;++i) {
while(head < tail && i - que[head] >= K) head++;
int sz = tail - head, x = A[i];
if(sz&1) x^=1;
if(!x) {
if(i + K > n) return -1;
res++;
que[tail++] = i;
}
}
return res;
}
};