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

文章目录

- 1. 题目
- 2. 解题

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]++;//左端点+1ans++;flip[i+K]--;//差分，右端点+1的位置 -1presum++;//前缀和更新}}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;}
};