Leetcode 995. K 连续位的最小翻转次数(困难)

最新推荐文章于 2024-09-30 14:51:11 发布

青春苦瓜

最新推荐文章于 2024-09-30 14:51:11 发布

阅读量414

点赞数

文章标签： leetcode 贪心算法算法

本文链接：https://blog.csdn.net/qq_41214789/article/details/121331838

版权

题意：
在仅包含 0 和 1 的数组 A 中，一次 K 位翻转包括选择一个长度为 K 的（连续）子数组，同时将子数组中的每个 0 更改为 1，而每个 1 更改为 0。
返回所需的 K 位翻转的最小次数，以便数组没有值为 0 的元素。如果不可能，返回 -1。
思路：
贪心思想，我们从左往右遍历数组，如果当前位置是0则翻转从这个位置开始长度为k的子数组，这样后来的翻转就不会影响前边已经翻转好了的元素，但是如果我们真正翻转的话，会超时，所以我们借助滑动窗口，用队列来模拟滑动窗口，队列中记录翻转的起始下标，对于后来的元素，我们根据队列中元素的个数即可知道当前元素被翻转了多少次，并根据结果确定当前元素是否需要再次翻转，如果需要，就加入队列并记录次数（但是每次要先把已经失效的起始下标移出队列，何为失效？就是起始+k影响不到当前）

public int minKBitFlips(int[] nums, int k) {
        Queue<Integer> queue=new LinkedList<>();
        int ans=0;
        int n= nums.length;
        for(int i=0;i<n;i++){
            if(!queue.isEmpty()&&queue.peek()+k<=i){//失效元素移出
                queue.poll();
            }
            if(queue.size()%2==nums[i]){//需要翻转
                if(i+k>n){//无法全1
                    return -1;
                }
                queue.add(i);
                ans++;
            }
        }
        return ans;
    }