滑動窗口的最大值

牛客上看到這樣一道題:

題目描述

給定一個數組和滑動窗口的大小,找出所有滑動窗口裏數值的最大值。例如,如果輸入數組{2,3,4,2,6,2,5,1}及滑動窗口的大小3,那麼一共存在6個滑動窗口,他們的最大值分別爲{4,4,6,6,6,5}; 針對數組{2,3,4,2,6,2,5,1}的滑動窗口有以下6個: {[2,3,4],2,6,2,5,1}, {2,[3,4,2],6,2,5,1}, {2,3,[4,2,6],2,5,1}, {2,3,4,[2,6,2],5,1}, {2,3,4,2,[6,2,5],1}, {2,3,4,2,6,[2,5,1]}。


解題思路

看到這種滑動窗口, 有進有出的題目,很容易的想到應該考察的是隊列的應用,而這道題很明顯要用到雙向隊列。因爲題目要求的是每個滑動窗口中的最大值,難點在於如何保存最大值,避免重複查找最大值,有一點動態規劃的意思。 我們可以在隊列頭部(右邊)保存最大值的index,每次讀入一個數,保證兩個條件:1、隊列左邊的index不過期,即在窗口內  2、隊列右邊保存的是窗口的最大值


下面是代碼:

class Solution  
{  
    public:  
        vector<int> maxInWindows(const vector<int>& num, unsigned int size)  
        {  
            vector<int> ans;  
            unsigned int len = num.size();  
            if(len>=size && size>=1)  
            {  
                deque<int> Q;  
                for(unsigned int i = 0; i<size; ++i)  
                {  
                    while(!Q.empty() && num[Q.back()]<=num[i])  
                        Q.pop_back();  
                    Q.push_back(i);  
                }  
                for(unsigned int i = size; i<len; ++i)  
                {  
                    ans.push_back(num[Q.front()]);  
                    while(!Q.empty() && num[Q.back()]<=num[i])  
                        Q.pop_back();  
                    if(!Q.empty() && Q.front()<=(int)(i-size))  
                        Q.pop_front();  
                    Q.push_back(i);  
                }  
                ans.push_back(num[Q.front()]);  
            }  
            return ans;  
        }  
};  



發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章