LeetCode 995. K 连续位的最小翻转次数（差分思想）

1. 题目

在仅包含 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
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

2. 解题

类似题目：
LeetCode 1094. 拼车
LeetCode 370. 区间加法（差分思想）
LeetCode 732. 我的日程安排表 III（差分思想）
LeetCode 1674. 使数组互补的最少操作次数（差分思想）

• 差分思想，用差分数组记录区间翻转情况
• 左端点翻转一次的话，长度为 K 的区间左端点+1，右端点+1的位置 -1
• 差分数组的前缀和为每个位置的翻转次数，翻转次数为偶数的话，状态不变

class Solution {
public:
    int minKBitFlips(vector<int>& A, int K) {
        int n = A.size();
        vector<int> flip(n+1, 0);
        int presum = 0;//每个位置的翻转次数
        int ans = 0;
        for(int i = 0; i <= n-K; ++i)
        {   //前面可以翻的位置
            presum += flip[i];
            if((A[i]+presum)%2 == 1)//不用再翻了
                continue;
            else
            {
                flip[i]++;//左端点+1
                ans++;
                flip[i+K]--;//差分，右端点+1的位置 -1
                presum++;//前缀和更新
            }
        }
        for(int i = n-K+1; i < n; i++)
        {   //剩余的不能再翻了，不足K个
            presum += flip[i];
            if((A[i]+presum)%2 == 0)
                return -1;//有不为1的，不能满足全部翻成1
        }
        return ans;
    }
};

224 ms 74.4 MB C++

---

我的CSDN博客地址 https://michael.blog.csdn.net/

长按或扫码关注我的公众号（Michael阿明），一起加油、一起学习进步！