[leetcode] Minimum Number of K Consecutive Bit Flips

我永远都学不会贪心算法罢了(╯‵□′)╯︵┻━┻

 

用队列flip_records记录翻转情况,flip_records中每个int元素记录一次翻转操作的起始位置。i.e. 元素值为start代表A[start]到A[start+k-1]的牌被翻转过。

从左到右遍历A中所有的牌,对每个当前遍历到的位置i:

    1. 检查flip_records队列中最前的位置记录(记为front),如果front+k-1<i,说明这次最早的翻转操作对当前的i没有影响,直接把这条记录从flip_records中弹出。

    2. 用bool变量need_flip表示当前是否需要增加(以i为起始位置的)翻转操作,need_flip的值可以这样计算:

        2-1. 因为先前做过弹出检查,所以当前flip_records中的翻转记录肯定都是对牌i有影响的。如果牌i在先前被翻转过偶数次则相当于没有翻转过。

               所以,用has_flipped记录牌“实际上相当于有没有翻转过”,has_flipped由flip_records队列中元素数量的奇偶决定。

        2-2. 然后,根据A[i]和has_flipped的值计算need_flip:

               //A[i]为0,实际上相当没有翻转过,则需要翻转,etc。

A[i] has_flipped need_flip
0 0 1
0 1 0
1 0 0
1 1 1

 

         2-3. 如果need_flip为真,说明牌i需要翻转。首先检查这次翻转会不会越界,如果越界说明不能成功,直接返回-1;如果可以翻转,递增翻转次数count。

最后返回翻转次数count。 

class Solution {
public:
    int minKBitFlips(vector<int>& A, int K) {
        int count=0, A_length=A.size();
        queue<int> flip_records;
        bool has_flipped, need_flip;
        for (int i=0;i<A_length;i++) {
            if (!flip_records.empty()) {
                if (flip_records.front()+K-1<i)
                    flip_records.pop();
            }
            has_flipped=flip_records.size()%2?true:false;
            need_flip=!A[i]^has_flipped;
            if (need_flip) {
                if (i+K>A_length)
                    return -1;
                flip_records.push(i);
                ++count;
            }
        }
        return count;
    }
};

 

posted @ 2019-03-03 12:36  丹尼尔奥利瓦  阅读(294)  评论(0编辑  收藏  举报