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. 解题

  • 差分思想,用差分数组记录区间翻转情况
  • 左端点翻转一次的话,长度为 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阿明),一起加油、一起学习进步!
Michael阿明

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章