Leetcode 955 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]

思路

采用滑动窗口方法,使用队列模拟滑动窗口,该滑动窗口的含义是指队列前K-1元素中,哪个元素对子区间进行了翻转。滑动窗口从左向右,如果当前位置i翻转,则将i放入队列中。遍历到新位置j(j<i+K),则此时队列的长度表示当前位置j被前K-1个元素的翻转次数。

  • 当 A[i] 为 0,如果 i 位置被翻转了偶数次,那么翻转后仍是 0,当前元素需要翻转;
  • 当 A[i] 为 1,如果 i 位置被翻转了奇数次,那么翻转后变成 0,当前元素需要翻转。

由上可知,如果 queue.size()%2==A[i],则当前元素需要翻转,且当i+K>N时,表示无论如何翻转都不可能为1 。

#include <bits/stdc++.h>
using namespace std;
class Solution
{
public:
    int minKBitFlips(vector<int> &A, int K)
    {
        int len = A.size();
        queue<int> q;
        int res = 0;
        for (int i = 0; i < len; i++)
        {
            // 当前滑动窗口滑动过程中要保持新位置j<i+K 保持K长度  超出需要删除队首元素
            if (!q.empty() && i >= q.front() + K)
            {
                q.pop();
            }
            if (q.size() % 2 == A[i])
            {
                if (i + K > len)
                {
                    return -1;
                }
                q.push(i);
                res++;
            }
        }
        return res;
    }
};
原文地址:https://www.cnblogs.com/dlvguo/p/14417631.html