题目描述
在仅包含 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]来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/minimum-number-of-k-consecutive-bit-flips
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解题思路
需要明白的一点是当遍历到第 i 个元素时,A[i]的状态是 0 还是 1,只和前面 k-1 个元素中,有多少个元素会使得 第 i 个元素进行翻转。
- 1.当前面有偶数个元素使得 A[i] 翻转,A[i] 不会改变, 当此时 A[i]==0 时,A[i] 需要翻转
- 2.当前面有奇数个元素使得 A[i] 翻转,A[i] 此时为 1,当此时 A[i] == 1时, A[i] 原来为 0 ,A[i] 需要翻转
现在只需要前面在翻转的时候,就保存下来,使用队列,保证按顺序最多只保存 k 个元素。
- i + k > len(A) 时,说明已没有足够的元素去翻转当前的 A[i],即不存在全部翻转为 1 的情况
代码
class Solution {
public:
int minKBitFlips(vector<int>& A, int K) {
queue<int> qindex;
int size = A.size();
int res=0;
for(int i=0;i<size;i++){
if(!qindex.empty() && i-K>=qindex.front()){
qindex.pop();
}
if(qindex.size()%2==A[i]){
if(i+K>size){
return -1;
}
qindex.push(i);
res += 1;
}
}
return res;
}
};
复杂度分析
- 时间复杂度:O(n)
- 空间复杂度:O(k)