LeetCode 995 Minimum Number of K Consecutive Bit Flips【思维】

传送门
题意: 给定一个长度为n的01数组, 每次可以将连续的k位翻转(o -> 1, 1 - > 0), 问能否将整个数字变成全1，能的话最少翻转多少次.

思路: 很明显可以分析出的是, 一但遇到0了，那么一定要翻转, 并且以后都不能再翻转它了, 也就是跟着翻转下去, 所以一个很暴力的想法就出来, 直接每次暴力翻转连续的k位, 那么复杂度是接近于n^2的, 所以我们就不写了, 那么我们再次分析得, 每次翻转一个k位后, 可以记录下一次我们要遇到的就是1，因为之前翻转过, 遇到的1就是0, 但是这里要判断是否在k位之内. 即我们每次翻转后并不暴力去更新, 而是记录下次遇到那一位时再次翻转… 详情看代码, 此时复杂度位O(n).
AC Code

class Solution {
public:
    int minKBitFlips(vector<int>& A, int K) {
        int cur = 0, ans = 0;
        for (int i = 0 ; i < A.size() ; ++ i) {
            if (i >= K && A[i-K] == 2) ++ cur;  
            // 如果超过k位, 那么下次要判断的就又要翻转
            if (cur % 2 == A[i]) {  // 遇到该翻转的了, 执行翻转
                if (i + K > A.size()) return -1;
                ++ cur; ++ ans;
                A[i] = 2;
                // 做好此位被翻转过的标记. 用于上面的判断是否在翻转的k位之内.
            }
        }
        return ans;
    }
};