在仅包含 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]
1.贪心:遍历一次A数组,遇到元素为0时就将该元素后面k位进行翻转
public int minKBitFlips(int[] A, int K) {
int n = A.length;
int ans = 0;
for(int i = 0; i < n; i++){
if(A[i]==0){
//保证后k位是有元素存在的
if(i+K>n){
return -1;
}
for(int j=i;j<i+K;j++){
A[j]^=1;
}
ans++;
}
}
return ans;
}
2.滑动窗口:在准备翻转A数组的第i个位置的元素前,需要对A数组中[i-k+1,i-1]的元素的翻转进行考虑。因为这一个窗口内每一个元素的翻转都对第i个位置的元素产生影响。将这个窗口内处理完后看需要对第i个位置元素翻转几次,对不同的次数进行不同的分析。cnt记录翻转次数(并不需要将窗口中的元素真的去翻转):
- 当前A[i]的值是1,然后被翻转奇数次,A[i]变为0,所以要转化为1,则需要cnt++;
- 当前A[i]的值是0,然后被翻转奇数次,A[i]变为1,不需要处理;
- 当前A[i]的值是1,然后被翻转偶数次,A[i]不变还是1,不需要处理
- 当前A[i]的值是0,然后被翻转偶数次,A[i]不变还是0,所以要转化为1,则需要cnt++
- 当前A[i]的值是0,但是该位置后面已经没有k位数字,则返回-1;
public int minKBitFlips(int[] A, int K) {
if(A == null || A.length == 0 ){
return 0;
}
int n = A.length;
Deque<Integer> q = new ArrayDeque<>();
int cnt = 0;
for(int i = 0;i < n ; i++){
//最左边元素索引已经不在窗口内
if(!q.isEmpty() && i-q.peekFirst()+1>K){
q.pollFirst();
}
//满足以下条件时,第i个元素需要翻转
if(q.size()%2==0 && A[i] == 0 || q.size()%2==1&&A[i]==1){
if(i+K>n){
return -1;
}
cnt++;
//第i个位置加入队列,表明第i个位置上的数反转了
q.addLast(i);
}
}
return cnt;
}
代码的关键在于使用了队列计算翻转的次数