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

leetcode 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]

提示：
1 <= A.length <= 30000
1 <= K <= A.length

尝试

尝试固定K位的大小去扫描数组，贪心的将每次扫到的K位的首位想办法置1。
思路基本没问题了，O(nk)的时间复杂度直接在第92个样例超时。

class Solution {
public:
    int minKBitFlips(vector<int>& A, int K) {
        int n = A.size();
        if(n == 0){
            return -1;
        }
        int ans = 0;
        int leftBound = 0,rightBound = 0;
        while(rightBound < n){
            if(rightBound - leftBound + 1 == K && A[leftBound] == 0){
                for(int i = leftBound ; i <= rightBound ; ++i){
                    A[i] = (A[i] + 1) % 2;
                }
                ans++;
                leftBound++;
            }else if(rightBound - leftBound + 1 == K){
                leftBound++;
            }
            rightBound++;
        }
        for(auto i : A){
            if(i == 0){
                ans = -1;
            }
        }
        return ans;
    }
};

题解

差分数组，之前的贪心思路仍然保持不变。
差分数组的目标是通过额外的数组来标记数组被翻转的情况，这里参考了官方题解的用法，用数组diff[i]来表示第A[i]被翻转的次数比A[i-1]多了几次，初始全为0。
然后另取变量reverseCount累加diff数组，这样对于当前遍历到的元素A[i]而言，当前的reverseCount就可以表示A[i]已经被翻转的次数。
譬如数组[0,0,1,0,1,1,0],K=3 。我们在i = 0 时翻转了数组,则i在0~2范围内的所有元素都被翻转了一次，在遍历这段元素的时候我们的reverseCount保持1，而diff[3]–(此时为-1，表示A[3]比左侧元素少转了一次，也可以说是为了抵消此前翻转导致reverseCount++所设置的还原)，我们将这次翻转称为第一次翻转。
当遍历到i = 2时，由于原先的元素为1，又已经被翻转了一次(表现为(A[2] + reverseCount) % 2 == 0),所以这里要对2~4范围内的所有元素再进行一次翻转，则此时reverseCount为2（即A[2]被翻转了两次），diff[5]–，我们将这次翻转成为第二次翻转。
遍历到i = 3 时，由于已经离开了第一次翻转的范围，而还落在第二次翻转的范围，所以此时reverseCount为2 + diif[3]即1，说明A[3]被翻转了一次，之后重复上述过程即可。
最后如果剩余元素不足K个却还需要再翻转的情况则说明无法再翻转，返回-1。

所以差分数组的核心就是用来标记范围，并在离开范围的第一时间通过减操作进行还原。

class Solution {
public:
    int minKBitFlips(vector<int>& A, int K) {
        int n = A.size();
        if(n == 0){
            return -1;
        }
        int ans = 0;
        int reverseCount = 0;
        vector<int> diff(n+5,0);
        for(int i = 0 ; i < n ; ++i){
            reverseCount += diff[i];
            if((reverseCount + A[i]) % 2 == 0){
                if(n - i < K){
                    ans = -1;
                    break;
                }
                ans++;
                reverseCount++;
                diff[i + K]--;
            }
        }
        return ans;
    }
};