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

题目链接

思路

直接模拟是会产生超时的，所以我们使用队列模拟滑动窗口，该滑动窗口的含义是前面 K - 1 个元素中，以哪些位置起始的 子区间进行了翻转。该滑动窗口从左向右滑动，如果当前位置 i 需要翻转，则把该位置存储到队列中。遍历到新位置 j (j < i + K)时，队列中元素的个数代表了 i 被前面 K - 1 个元素翻转的次数。

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

即 **que.size()%2==A[i]**当前元素需要翻转。注意：当 i + K > N 时，说明需要翻转大小为 K 的子区间，但是后面剩余的元素不到 K 个了，所以返回 -1。

代码

class Solution {
    public int minKBitFlips(int[] A, int K) {
        int n=A.length;
        Deque<Integer> window = new LinkedList<>();
        int res=0;
        for (int i = 0; i <n ; i++) {
            if(!window.isEmpty()&&i>=window.peek()+K) window.pop();//说明这个i位置的元素已经超出队列中首位位置所形成的K大小的翻转范围了，所以要处理，让队列首出队列
            if(window.size()%2==A[i]){//符合上述情况需要反转，不等说明满足不需要翻转
                if(i+K>n) return -1;//如果剩余元素已经不构成K个，无法反转
                window.add(i);
                res+=1;//记录翻转次数
            }
        }
        return res;
    }
}