用一个队列q来表示这个数字之前与之相关的数字(就是比如说A=【0,0,1】,K=2,那么当第0位数字0进行翻转时,第2位数字1并不会翻转,这样第0位和第2位就是不相关的数字)翻转了几次(如果该数字进行翻转,则把其加入队列,然后用q.size()可以表示翻转了几次,如果q的最前面的数字+K>i,表示此时q的最前面的数字已经不会影响后面的数字了,此时可以删掉),如果翻转了奇数次并且A[i]=1,那么此时该数字也需要翻转,同理,如果翻转了偶数次并且A[i]=0,那么此时该数字也需要翻转。
然后当i>n-K以后,数字已经无法翻转了,此时如果遇到需要翻转的数字直接返回-1即可。
class Solution {
public:
int minKBitFlips(vector<int>& A, int K) {
int res = 0, n = A.size();
queue<int> q;
for(int i=0;i<n;i++){
if(!q.empty()&&(i>=q.front()+K)){
q.pop();
}
if(q.size()%2==A[i]){
if(i+K>n){
return -1;
}
q.push(i);
res++;
}
}
return res;
}
};