首页 > 编程语言 >[思维提升|干货All in]6种算法解决LeetCode困难题:滑动窗口最大值

[思维提升|干货All in]6种算法解决LeetCode困难题:滑动窗口最大值

时间:2023-03-14 13:44:49浏览次数:52  
标签:vector nums int res 最大值 st 干货 maxn LeetCode

为了更好的阅读体验,欢迎阅读原文:
[思维提升|干货All in]6种算法解决LeetCode困难题:滑动窗口最大值 (eriktse.com)

最近在leetcode遇到一道非常经典的题目:239. 滑动窗口最大值 - 力扣(LeetCode)

image

以前只会看题解用单调队列做,最近研究一下发现是一道很好的题,可以帮助我们提升“维护区间最值”的算法思维。

先介绍一下我解决这题所用的算法及其复杂度:

  • 单调队列 O(n)
  • st表 O(nlogn)
  • 树状数组 O(n(logn)^2)
  • 多重集合法 O(nlogn)
  • 莫队O (n sqrt{n})
  • 优先队列 O(nlogn)

首先确定一点,单调队列是解决这道题最好的办法,但是其他的方法的适用范围更广。

1、单调队列

首先可以参考几篇优秀的文章:

算法学习笔记(66): 单调队列 - 知乎 (zhihu.com)

单调队列 - OI Wiki (oi-wiki.org)

我这里提几点值得注意的地方:

1.单调队列中存放的是下标,而不是元素值

2.单调队列是一个双端队列,尾插前先查队头后查队尾

3.单调队列维护的是元素值的单调性

有了这几点注意,代码就很好写了:

class Solution {
public:
    static const int maxn = 1e5 + 9;
    int a[maxn];
    deque<int> dq;
    vector<int> maxSlidingWindow(vector<int>& nums, int k) {
        vector<int> res;
        int n = nums.size();
        for(int i = 1;i <= n; ++ i)a[i] = nums[i - 1];
        for(int i = 1;i <= n; ++ i)
        {
            int x = a[i];
            while(!dq.empty() and dq.front() < i - k + 1)dq.pop_front();
            while(!dq.empty() and x >= a[dq.back()])dq.pop_back();
            dq.push_back(i);
            if(i >= k)res.push_back(a[dq.front()]);
        }
        return res;
    }
};

我做题习惯把输入数组存一个array,大家请勿介意。

2.st表

st表是一种基于DP(动态规划)思想的算法,也算是一种数据结构吧。

st表可以静态维护区间的最值,需要用的时间来预处理,后可以O(1)查询。

我们定义dp[i][j]表示
表示区间的最值,在这道题里我们认为是最大值(维护最小值同理)。看一下能不能开下,大概是maxn * 20的大小,可以开下。

通过定义不难发现,dp[i][0] = a[i],因为此时区间长度为1,那么最值就是元素a[i]本身。当j增大时,我们有转移方程(具体的原因可简单自行推导,以后的文章中也会有讲解):

dp[i][j] = max(dp[i][j - 1], dp[i + 2^(j-1)][j - 1])

查询十分方便,方法是从两边往中间尽可能多地覆盖。假设要查询的区间是[l,r]
,那么我们可以得到区间长度r - l + 1,现在求一个比长度小且最大的2的幂,然后把左右两块取大即可。

直接看代码:

class Solution {
public:
    static const int maxn = 1e5 + 9;
    int a[maxn], st[maxn][30];//st[i][j]表示[i, i + 2^j - 1]的最大值
    int queMax(int l, int r)
    {
        int k = log(r - l + 1) / log(2);
        return max(st[l][k], st[r - (1<<k) + 1][k]);
    }
    vector<int> maxSlidingWindow(vector<int>& nums, int k)
    {
        memset(st, 0xcf, sizeof st);
        vector<int> res;
        int n = nums.size();
        for(int i = 1;i <= n; ++ i)a[i] = nums[i - 1];
        for(int i = 1;i <= n; ++ i)st[i][0] = a[i];
        for(int k = 1;k <= 20; ++ k)
        {
            for(int i = 1;i <= n and i + (1 << (k - 1)) <= n; ++ i)
            {
                st[i][k] = max(st[i][k - 1], st[i + (1 << (k - 1))][k - 1]);
            }
        }
 
        for(int i = 1;i + k - 1 <= n; ++ i)res.push_back(queMax(i, i + k - 1));
 
        return res;
    }
};

3、树状数组

看见树状数组可能有小朋友会感到疑惑了哦,树状数组不是维护区间和的吗?怎么还来凑“区间最值”的热闹了。

其实树状数组不仅可以维护区间和,还可以“动态维护区间最值”,但是维护的方法和区间和略有不同。这一次主要看一下代码吧,具体的原理之后再讲。

树状数组节点t[k]维护的是区间[k - lowbit(k) + 1, k]。

树状数组主要突出的优点就是可以动态维护,但是注意在维护区间最值的时候仅可单点修改,不支持区间修改。

class Solution {
public:
    static const int maxn = 1e5 + 9;
    //树状数组
    int a[maxn], t[maxn], n;//t[i] 表示 a[i - lowbit(i) + 1] ~ a[i]的最大值
    int lowbit(int x){return x & -x;}
 
    void update(int k, int x)// modify a[k] to x
    {
        a[k] = x;
        while(k <= n)
        {
            t[k] = x;
            for(int i = 1;i < lowbit(k); i <<= 1)t[k] = max(t[k], t[k - i]);
            k += lowbit(k);
        }
    }
 
    int queMax(int l, int r)
    {
        int res = a[r];//单点查询
        while(l <= r)
        {
            for(;r - lowbit(r) >= l; r -= lowbit(r))res = max(res, t[r]);
            res = max(res, a[r --]);
        }
        return res;
    }
 
    vector<int> maxSlidingWindow(vector<int>& nums, int k)
    {
        n = nums.size();
        for(int i = 1;i <= n; ++ i)a[i] = nums[i - 1];
        vector<int> res;
        for(int i = 1;i <= n; ++ i)
        {
            update(i, a[i]);
            if(i >= k)res.push_back(queMax(i - k + 1, i));
        }
 
        return res;
    }
};

4、多重集合法

这种方法就十分简单粗暴了,就是维护一个不断移动的multiset,简直是暴力之王。

class Solution {
public:
 
    vector<int> maxSlidingWindow(vector<int>& nums, int k)
    {
        multiset<int> st;
        vector<int> res;
        for(int i = 0;i < k; ++ i)st.insert(nums[i]);
        res.push_back(*st.rbegin());
        for(int i = k;i < nums.size(); ++ i)
        {
            st.erase(st.find(nums[i - k]));
            st.insert(nums[i]);
            res.push_back(*st.rbegin());
        }
 
        return res;
    }
};

5、莫队

莫队需要基于multiset,在这道题里的优势并不明显,因为这题的询问都是有顺序的,但是可以写个莫队练个手。

莫队在处理随机区间查询问题的时候有独特的优势,因为足够暴力,所以维护的东西可以很多很杂,比如区间和,区间最值,区间元素种类数等。

以后我会详细讲莫队的,欢迎大家访问我的个人博客:https://www.eriktse.com

在右上角留下邮箱订阅我的博客,每周更新优质的算法/技术/互联网文章!

class Solution {
public:
    static const int maxn = 1e5 + 9;
    int a[maxn], pos[maxn], ans[maxn], n;
    struct Q
    {
        int l, r, id;//询问离线
    }q[maxn];//outline algorihm
 
    multiset<int> st;
 
    void Add(int k)//把a[k]加入到区间内
    {
        st.insert(a[k]);
    }
 
    void Del(int k)
    {
        st.erase(st.find(a[k]));
    }
 
    vector<int> maxSlidingWindow(vector<int>& nums, int k)
    {
        n = nums.size();
        for(int i = 1;i <= n; ++ i)a[i] = nums[i - 1];
        int siz = sqrt(n);
        for(int i = 1;i <= n; ++ i)pos[i] = i / siz;
        int m = n - k + 1;
        for(int i = 1;i <= m; ++ i)q[i].l = i, q[i].r = i + k - 1, q[i].id = i;
 
        sort(q + 1, q + 1 + m, [this](const Q &u, const Q &v)
        {
            return pos[u.l] == pos[v.l] ? u.r < v.r : pos[u.l] < pos[v.l];
        });
        int L = 1, R = 0;//当前区间
        for(int i = 1;i <= m; ++ i)
        {
            //[L, R] -> [l, r]
            int l = q[i].l, r = q[i].r, id = q[i].id;
            while(L < l)Del(L ++);
            while(L > l)Add(-- L);
            while(R > r)Del(R --);
            while(R < r)Add(++ R);
            ans[id] = *st.rbegin();
        }
        vector<int> res;
        for(int i = 1;i <= m; ++ i)res.push_back(ans[i]);
        return res;
    }
};

6、优先队列

优先队列可以理解为一个“堆”结构。

我们知道优先队列可以维护最值,但是他只有一个堆顶怎么办呢?

我们只能保证堆顶是最大值但是却无法保证堆顶是在窗口内的呀!

为了解决这个问题,我们在每一次查询堆顶之前,都要对堆顶进行检查,直到堆顶在窗口内才能输出。

注意以下几点:

1.堆里存放的是下标,但是比较函数用值比较。

2.每次取出元素之前堆顶检查,只要堆顶的位置不在窗口内就一直弹出。

上代码!

const int maxn = 1e5 + 9;
int a[maxn];
class Solution {
public:
    struct cmp{
        bool operator ()(const int &u, const int &v)const
        {
            return a[u] < a[v];
        }
    };
    priority_queue<int, vector<int>, cmp> pq;
    vector<int> maxSlidingWindow(vector<int>& nums, int k)
    {
        int n = nums.size();
        for(int i = 1;i <= n; ++ i)a[i] = nums[i - 1];
 
        vector<int> res;
        for(int i = 1;i <= n; ++ i)
        {
            pq.push(i);
            while(!pq.empty() and pq.top() < i - k + 1)pq.pop();
            if(i >= k)res.push_back(a[pq.top()]);
        }
 
        return res;
    }
};

标签:vector,nums,int,res,最大值,st,干货,maxn,LeetCode
From: https://www.cnblogs.com/eriktse/p/17214664.html

相关文章

  • 32位汇编语言实现求数组的最大值
    INCLUDEIrvine32.inc.dataarrdd99,2,3,1,22,188,7,77,54,10;定义数组 lendd($-arr)/4;用当前地址减去数组首元素地址除以4得到数组的长度.codemainPROC......
  • [LeetCode] 62. 不同路径 java 动态规划解法
    classSolution{publicintuniquePaths(intm,intn){//确定dp数组以及下标的含义//dp[i][j]:表⽰从(0,0)出发,到(i,j)有dp[i][j]条不同的路径......
  • 电子行业干货:SMT、PCBA、PCB的区别和联系
    电子行业的新人,经常会将PCBA、SMT、PCB这三个完全不同的东西混淆,今天就通过简单明了的语言来谈谈PCBA、SMT、PCB三者之间区别与联系—————————————————⭐PC......
  • LeetCode142. 环形链表 II
    题目描述:给定一个链表的头节点 head ,返回链表开始入环的第一个节点。 如果链表无环,则返回 null。如果链表中有某个节点,可以通过连续跟踪next指针再次到达,则链表中......
  • # yyds干货盘点 # 已知中心点的经纬度和长宽,怎么求矩形左上角和右下角经纬度呀?
    大家好,我是皮皮。一、前言前几天在Python最强王者交流群【缄默】问了一个经纬度处理的问题,这里拿出来给大家分享下。二、实现过程这个问题确实挺难的,像是数学题目,这里【吴超......
  • 【LeetCode回溯算法#11】解数独,这次是真的用回溯法处理二维数组
    解数独力扣题目链接(opensnewwindow)编写一个程序,通过填充空格来解决数独问题。一个数独的解法需遵循如下规则:数字1-9在每一行只能出现一次。数字1-9在每一列只......
  • #yyds干货盘点#简单的vuex实现
    实现一个vuex插件​​pvuex​​ 初始化:Store声明、install实现,vuex.js:letVue;//定义install方法完成赋值与注册functioninstall(_Vue){Vue=_Vue;Vue.mixin({......
  • #yyds干货盘点#Vuex中的store
    每一个​​Vuex​​应用的核心就是store(仓库)。store基本上就是一个容器,包含着应用中大部分的state(状态)。​​Vuex​​和单纯的全局对象有以下两点不同:​​Vuex​​ 的状态......
  • [Leetcode Weekly Contest]336
    链接:LeetCode[Leetcode]2586.统计范围内的元音字符串数给你一个下标从0开始的字符串数组words和两个整数:left和right。如果字符串以元音字母开头并以元音字母结......
  • LeetCode 3.无重复字符的最长子串
    题目链接在这里:3.无重复字符的最长子串-力扣(LeetCode)这道题学习了几何函数set()的用法1classSolution(object):2deflengthOfLongestSubstring(self,s:st......