LeetCode 995. K 连续位的最小翻转次数

995. K 连续位的最小翻转次数

难度:困难

在仅包含 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]

 

【分析】

从左向右，每次遇到0就翻转k位。当A[n-K]是1时，此时数组全部为1则翻转总次数即为所求；若此时还有0则无解。

但是该方法算法复杂度高，因为当数组是类似1,0,1,0,1交错排列时.......，复杂度是O(nK)。

数组中的数翻转两次等于没翻转。

当我们考虑第i位数是否需要翻转时，和i之前K-1位有关：

if A[i]时0，且第i位已经翻转了偶数次，则翻转后第i位还是0，当前需要翻转；if A[i]时1，且第i位已经翻转了奇数次，则翻转后第i位变成1，当前需要翻转。

现在需要求第i位已经翻转的次数：

我们维护一个队列queue，然后遍历数组A，每当一个元素需要翻转时，我们就将元素索引放入队列；

假设队列中第一个元素为i，则当遍历到第i+K个元素时，我们就将队列首元素出队。

这样维护的队列中元素的个数即当前第i个元素已经翻转的次数。我们还能得到公式len(queue)%2==A[i]时，此时需要翻转。

【代码】

class Solution {
    public int minKBitFlips(int[] A, int K) {
        Queue<Integer> queue = new ArrayDeque<>();
        int sumFlips = 0;
        int len = A.length;
        for (int i = 0; i < len; i++) {
            if (!queue.isEmpty() && queue.peek() == i-K) {
                queue.poll();
            }
            if (queue.size() % 2 == A[i]) {
                if (len - K < i) {
                    return -1;
                }
                sumFlips++;
                queue.offer(i);
            }
        }
        return sumFlips;
    }
}

【复杂度】

时间复杂度O(N)

空间复杂度O(K)