在仅包含 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
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/minimum-number-of-k-consecutive-bit-flips
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解题思路:
这道题是通过参考官方思路写出来的,当然官方思路不太容易理解,这里我用我自己理解的方式解释给大家,整体的思路就是贪心,当前是1不翻转,是0就翻转,一直到最后,如果在最后K个数范围内还有0,那没办法了,直接返回-1,每次翻转前要统计已经翻转的次数,这个次数并不是累加的,举个例子,如果k是2,第一个数是0,第二个数是1,那么首先翻转第一个K长度数组,变成了[1 ,0],这个时候统计翻转次数的reverse数组要更新,将 0 + 2 位置的值-1,意思是从这开始没有翻转,相对于前一个k范围数组少翻转一次,计算到这时要减去1,每翻转一次就要更新当前翻转次数,即 cur++,所以总而言之,i + k 位置相当于标记,指明从这开始少翻转 reverse[i + k] 次。
当然为了提高内存执行效率,可以不用reverse数组,那如何标记呢?这就要利用到题中所给的A数组,因为数组中值的范围为0或者1,所以只要把A数组中对应位置的值更新到大于1即可作为标记,详细思路见题解,代码如下:
class Solution {
public:
int minKBitFlips(vector<int> &A, int K) {
int len = A.size();
int count = 0;
int cur = 0;
// 翻转次数
vector<int> reverse(len + 1);
for(int i = 0; i < len; i ++) {
// 统计当前翻转次数
cur += reverse[i];
// 说明翻转过了
if((cur + A[i]) % 2 == 0) {
// 超过范围
if(i + K > len) {
return -1;
}
count ++;
cur ++;
reverse[i + K] --;
}
}
return count;
}
};
/*作者:heroding
链接:https://leetcode-cn.com/problems/minimum-number-of-k-consecutive-bit-flips/solution/zui-xiang-xi-cti-jie-by-heroding-1c4p/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。*/