【LeetCode】K 连续位的最小翻转次数Java题解

114 阅读2分钟

一起养成写作习惯!这是我参与「掘金日新计划 · 4 月更文挑战」的第13天,点击查看活动详情

题目描述

给定一个二进制数组 nums 和一个整数 k 。

k位翻转 就是从 nums 中选择一个长度为 k 的 子数组 ,同时把子数组中的每一个 0 都改成 1 ,把子数组中的每一个 1 都改成 0 。

返回数组中不存在 0 所需的最小 k位翻转 次数。如果不可能,则返回 -1 。

子数组 是数组的 连续 部分。


示例 1:

输入:nums = [0,1,0], K = 1
输出:2
解释:先翻转 A[0],然后翻转 A[2]。
示例 2:

输入:nums = [1,1,0], K = 2
输出:-1
解释:无论我们怎样翻转大小为 2 的子数组,我们都不能使数组变为 [1,1,1]。

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/minimum-number-of-k-consecutive-bit-flips
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

通过代码

public class DayCode {
    public static void main(String[] args) {
        int[] A = new int[]{0, 1, 0};
        int K = 1;
        int ans = new DayCode().minKBitFlips(A, K);
        System.out.println("ans is " + ans);
    }

    /**
     * https://leetcode-cn.com/problems/minimum-number-of-k-consecutive-bit-flips/
     * 时间复杂度O(n * k),空间复杂度O(1)
     * @param A
     * @param K
     * @return
     */
    public int minKBitFlips(int[] A, int K) {
        int ans = 0;
        int n = A.length;
        for (int i = 0; i < n; i++) {
            if (A[i] == 0) {
                if (i + K > n) {
                    return -1;
                }
                for (int j = i; j < i + K; j++) {
                    A[j] ^= 1;
                }
                ans++;
            }
        }
        return ans;
    }

    /**
     * https://leetcode-cn.com/problems/minimum-number-of-k-consecutive-bit-flips/
     * 时间复杂度O(n),空间复杂度O(n)
     * @param A
     * @param K
     * @return
     */
    public int minKBitFlipsTwo(int[] A, int K) {
        int ans = 0;
        int n = A.length;
        Deque<Integer> queue = new LinkedList<>();
        for (int i = 0; i < n; i++) {
            if (queue.size() > 0 && i > queue.peek() + K - 1) {
                queue.removeFirst();
            }
            if (queue.size() % 2 == A[i]) {
                if (i + K > n) {
                    return -1;
                }
                queue.add(i);
                ans++;
            }
        }
        return ans;
    }

}

思路分析

  • 这个题目含义容易理解,就是翻转K个连续位, 如果翻转后的结果,可以全部为1,即返回翻转次数,如果不能,则返回-1

  • 方法一是题目含义的具体实现,位运算很巧妙。

  • 方法二则是不真的翻转数组,需要深入的思考,应用了滑动窗口的思想。A[i] 翻转偶数次的结果是 A[i];翻转奇数次的结果是 A[i] ^ 1。

  • 每日算法坚持一题,加油!