牛客刷题之滑动窗口的最大值

最新推荐文章于 2025-02-12 15:44:59 发布

原创最新推荐文章于 2025-02-12 15:44:59 发布 · 212 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#队列 #数据结构 #算法

刷题记录专栏收录该内容

52 篇文章

订阅专栏

该博客介绍了如何利用双端队列（deque）在O(n)的时间复杂度内解决滑动窗口最大值的问题。两种解题方法被提出，包括暴力解法（O(nk)）和优化方法（O(n)）。优化方法中，双端队列用于存储窗口内的下标，通过不断更新队列头部和尾部，确保队列始终包含当前窗口内的最大值。这种方法提高了算法效率，适用于大数据量的数组处理。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目描述

给定一个数组和滑动窗口的大小，找出所有滑动窗口里数值的最大值。例如，如果输入数组{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]}。

牛客链接：

https://www.nowcoder.com/practice/1624bc35a45c42c0bc17d17fa0cba788?tpId=13&&tqId=11217&rp=1&ru=/ta/coding-interviews&qru=/ta/coding-interviews/question-ranking

解题思路：

用两种方法解决：
第一种暴力解决，两层循环找到最大值
第二种采用双端队列，队列中的头节点保存的数据比后面的要大。比如当前假如的数据比队尾的数字大，说明当前这个数字最起码在从现在起到后面的过程中可能是最大值，而之前队尾的数字不可能最大了，所以要删除队尾元素。此外，还要判断队头的元素是否超过size长度，由于存储的是下标，所以可以计算得到；在双端队列中保存的数字是传入的向量的下标；

代码一：
时间复杂度o(nk)

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

代码二：
时间复杂度o（n），空间复杂度为o（n）

class Solution {
public:
    vector<int> maxInWindows(const vector<int>& num, unsigned int size)
    {
        vector<int> result;
        int len = num.size();
        if(len < size || size< 1) return result;
        deque<int> maxIndex;
        for(int i=0; i<size; i++) {
            //当队列的末尾下标对应值小于当前下标对应值时，删除，一直循环
            while(!maxIndex.empty() && num[i]>num[maxIndex.back()]) {
                maxIndex.pop_back();
            }
            maxIndex.push_back(i);
        }
        for(int i=size; i<len; i++) {
            result.push_back(num[maxIndex.front()]);
            //当队列的末尾下标对应值小于当前下标对应值时，删除，一直循环
            while(!maxIndex.empty() && num[i]>=num[maxIndex.back()]) {
                maxIndex.pop_back();
            }
            //当队列的队首值已经超过滑动窗口大小，删除
            if(!maxIndex.empty() && (int)(i-size)>=maxIndex.front()) {
                maxIndex.pop_front();
            }
            maxIndex.push_back(i);
        }
        result.push_back(num[maxIndex.front()]);
        return result;
    }
};

注意：
双端队列