题目描述
在仅包含 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]。
贪心
这题我们遇见0就翻它,1就跳过.我们需要一个数组来记录翻K个长度的子区间的状态。当我们翻K个,那么 K+i个起始位置就比前面少翻一次,就减1。还有在这个过程中,我们需要一个cnt来连续的记录这个过程,通过它来判断当前的数值是否能被2整除 。
参考文献
C++ 代码
class Solution {
public:
int minKBitFlips(vector<int>& A, int K) {
int res = 0;
int cnt = 0;
int n = A.size(); //记录个数
vector <int> B(n + 1); // 单独开一个数组
for (int i = 0;i < n;i ++){ //遍历数组
cnt += B[i];
//已经翻过了
if ((cnt + A[i]) % 2 == 0){
if (i + K > n) return -1; // 超出范围
cnt ++;
res ++;// 答案的记录
B[i + K] --;//比之前少翻一遍
}
}
return res;
}
};