在仅包含 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.length <= 30000
1 <= K <= A.length
思路:本题参考了官方题解的做法。采用贪心思想:
如果最左边的元素是 0,那么我们一定要翻转从位置 0 开始的子数组。类似的,如果最左边的元素是 1,我们不应该翻转从位置 0 开始的子数组。这证明了我们可以贪心地执行这一过程:在明确是否要反转第一个子数组之后(位置 0 至 K-1),我们可以考虑将数组中第一个元素(值为 1)移除,然后重复这个过程。
答案的实现方式比较优美,具体代码如下:
class Solution {
public int minKBitFlips(int[] A, int K) {
int ans=0;
int n=A.length;
int[] hint=new int[n];
int flip=0;
for(int i=0;i<n;i++) {
flip^=hint[i];
if(A[i]==flip) {
ans++;
if(i+K>n) return -1;
flip^=1;
if(i+K<n) hint[i+K]^=1;
}
}
return ans;
}
}