滑動窗口的最大值(42)

題目

給定一個數組和滑動窗口的大小,找出所有滑動窗口裏數值的最大值。例如,如果輸入數組{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]}


方法一:

1、分析
使用暴力的方法來進行,分別求出每個子數組中的最大值。時間複雜度爲 O(nk)O(n*k) ,k爲滑動窗口的大小,n爲數組的大小。
2、代碼

class Solution {
public:
    vector<int> maxInWindows(const vector<int>& num, unsigned int size)
    {
        vector<int> res;
        unsigned int len=num.size();
        if(len<=0 ||size==0)
            return res;
        unsigned int count=len-size+1;
        for(unsigned int i=0;i<count;++i)
        {
            unsigned int end=i+size;
            int maxNum=num[i];
            for(unsigned int j=i+1;j<end;++j)
            {
                if(maxNum<num[j])
                    maxNum=num[j];
            }
            res.push_back(maxNum);
        }
        return res;
    }
};

方法二

1、分析
使用STL中的隊列 deque 來將最大值的下標保存下來,同時保證該隊列的首元素就是最大值的下標。

2、代碼

class Solution {
public:
    vector<int> maxInWindows(const vector<int>& num, unsigned int size)
    {
        vector<int> res;
        if(num.size()>=size && size>=1)
        {
            deque<int> dq;
            //求第一個滑窗中的size個元素的最大值
            for(unsigned int i=0;i<size;++i)
            {
                while(!dq.empty() && num[i]>=num[dq.back()])
                {
                    //pop_back之後,前面的元素就會變成末尾
                    dq.pop_back();
                }
                dq.push_back(i);
            }
            //接下來將後續添加的元素與dq中的進行比較,求最大值
            for(unsigned int j=size;j<num.size();++j)
            {
                res.push_back(num[dq.front()]);
                
                while(!dq.empty() && num[j]>=num[dq.back()])
                    dq.pop_back();
                if(!dq.empty() && dq.front()<=j-size)
                    dq.pop_front();
                dq.push_back(j);
            }
            res.push_back(num[dq.front()]);
        }
        return res;
    }
};
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章