【劍指Offer系列59】滑動窗口的最大值

題目

給定一個數組 nums 和滑動窗口的大小 k，請找出所有滑動窗口裏的最大值。

代碼

Python

class Solution:
    def maxSlidingWindow(self, nums: List[int], k: int) -> List[int]:
        deq=collections.deque() # 雙端隊列
        res,n=[],len(nums)
        for i,j in zip(range(1-k,n-k+1),range(n)): # 遍歷窗口左右邊界
            # 每輪窗口滑動移除了元素nums[i−1]，則將deq內的對應元素一起刪除,以保持deq內僅包含窗口內的元素
            if i>0 and deq[0]==nums[i-1]: deq.popleft() 
            # 刪除deq內所有小於nums[j]的元素，以保持deq遞減
            while deq and deq[-1]<nums[j]: deq.pop()
            deq.append(nums[j])
            # 若已形成窗口，將窗口最大值添加至列表res
            if i>=0: res.append(deq[0])
        return res

C++

class Solution {
public:
    vector<int> maxSlidingWindow(vector<int>& nums, int k) {
        deque<int> deq;
        vector<int> res;
        int n = nums.size();
        for (int i=0; i<n; i++) {
            while (!deq.empty() && nums[i]>nums[deq.back()]) {
                deq.pop_back();
            }   
            if (!deq.empty() && deq.front()<i-k+1) deq.pop_front();
            deq.push_back(i);
            if (i>=k-1) res.push_back(nums[deq.front()]);
        }
        return res;
    }
};