[leetcode] Minimum Number of K Consecutive Bit Flips

我永遠都學不會貪心算法罷了(╯‵□′)╯︵┻━┻算法

 

用隊列flip_records記錄翻轉狀況,flip_records中每一個int元素記錄一次翻轉操做的起始位置。i.e. 元素值爲start表明A[start]到A[start+k-1]的牌被翻轉過。spa

從左到右遍歷A中全部的牌,對每一個當前遍歷到的位置i:code

    1. 檢查flip_records隊列中最前的位置記錄(記爲front),若是front+k-1<i,說明此次最先的翻轉操做對當前的i沒有影響,直接把這條記錄從flip_records中彈出。blog

    2. 用bool變量need_flip表示當前是否須要增長(以i爲起始位置的)翻轉操做,need_flip的值能夠這樣計算:隊列

        2-1. 由於先前作過彈出檢查,因此當前flip_records中的翻轉記錄確定都是對牌i有影響的。若是牌i在先前被翻轉過偶數次則至關於沒有翻轉過。ip

               因此,用has_flipped記錄牌「實際上至關於有沒有翻轉過」,has_flipped由flip_records隊列中元素數量的奇偶決定。it

        2-2. 而後,根據A[i]和has_flipped的值計算need_flip:io

               //A[i]爲0,實際上至關沒有翻轉過,則須要翻轉,etc。table

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。class

最後返回翻轉次數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;
    }
};
相關文章
相關標籤/搜索