传送门算法
在仅包含 0 和 1 的数组 A 中,一次 K 位翻转包括选择一个长度为 K 的(连续)子数组,同时将子数组中的每一个 0 更改成 1,而每一个 1 更改成 0。
返回所需的 K 位翻转的最小次数,以便数组没有值为 0 的元素。若是不可能,返回 -1。数组
示例 1:编码
输入:A = [0,1,0], K = 1
输出:2
解释:先翻转 A[0],而后翻转 A[2]。code
示例 2:blog
输入:A = [1,1,0], K = 2
输出:-1
解释:不管咱们怎样翻转大小为 2 的子数组,咱们都不能使数组变为 [1,1,1]。图片
示例 3:ip
输入: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
提示:get
1 <= A.length <= 30000 1 <= K <= A.length
来源:力扣(LeetCode)it
int minKBitFlips(int* A, int ASize, int K){ }
题目要求就是对于一个已知的01组成的数组进行K长度的子数组的翻转(0->1,1->0),求翻转到最后全部元素都为1的最少次数,若是不能达到这个目的就返回-1;
这一题应用到了不少的方法:好比贪心算法,差分,滑窗。
咱们所要翻转的是一个长度为K的小窗口,可翻转,翻转两次等于没翻。
从作到有判断是否每一个元素都须要翻转。
若是须要翻转就记录翻转次数,记录就是差分遍历的首位(i)加,末位(i+k)减。最后再利用差分求累计翻转次数;
若是须要翻转,但剩余元素长度已经不足K个,那么就是没法完成全置1的目标,返回-1便可。
这里是对于差分数组的创建以及初始化,以及对最后须要返回的反转次数ans的定义,对于revCnt就是算做咱们的差分数组的首位diff[i];
int diff[ASize + 1]; memset(diff, 0, sizeof(diff)); int ans = 0, revCnt = 0;
for (int i = 0; i < ASize; ++i) { }
revCnt += diff[i];
if ((A[i] + revCnt) % 2 == 0) { if (i + K > ASize) { return -1; } }
++ans; ++revCnt; --diff[i + K];
int minKBitFlips(int* A, int ASize, int K) { int diff[ASize + 1]; memset(diff, 0, sizeof(diff)); int ans = 0, revCnt = 0; for (int i = 0; i < ASize; ++i) { revCnt += diff[i]; if ((A[i] + revCnt) % 2 == 0) { if (i + K > ASize) { return -1; } ++ans; ++revCnt; --diff[i + K]; } } return ans; }