单调栈浅学

最新推荐文章于 2024-08-15 22:41:27 发布

zhper

最新推荐文章于 2024-08-15 22:41:27 发布

阅读量293

点赞数 1

文章标签：算法 c++

本文链接：https://blog.csdn.net/zzzzzzyie/article/details/125585408

版权

这里给出三道单调栈的题目

对于单调栈的个人理解

给出一个数组 [2,1,2,4,3]，我们想找出每一个元素的下一个更大元素（从左往右），比如2的下一个更大元素是4，1的下一个更大元素是2…，但是4没有下一个更大元素，我们就用 -1 填充 , 同理3也没有
于是我们最后可以得到这个数组的下一个更大元素的元素数组 [4,2,4,-1,-1]，其对应的下标数组 [3,2,3,-1,-1]，我们可以怎样理解呢？

（图片来自与labuladong的力扣题解）

图片来自与labuladong的力扣题解

通过这张图应该能理解到了吧，下一个更大元素就是数组后面的把它挡住了的

如何代码实现呢

我也不知道怎么说，通过代码配上简单的注释，再仔细理解理解叭

//nums:给出的数组
//ans:记录答案
//stack<int> s : 储存单调序列
    for (int i = N; i >= 1; i--) {
        while (!s.empty() && nums[s.top()] <= nums[i]) {
            s.pop();
        }
        ans[i] = s.empty() ? 0 : s.top();
        s.push(i);
    }

这个算是单调栈的模板了，用一个stack s储存单调的序列，从后往前遍历数组，因为比他小的数没法挡住它，所以直接pop就行，知道遇到比他大的就是下一个更大的数
这里我个人推荐stack储存数组的下标，可操作性更大，当然对于个别题也可以直接储存元素值

循环数组

上面的数组从左到右找不到了就算了，但是循环数组允许我们后面的数字在它前面找到下一个更大的数字
图片来自与labuladong的力扣题解
首先我们可以copy一份数组，将这个数组变为两份，这样也相当于一个循环数组了，但是题目给的数组可能不会让你过，这时我们想，我们为什么要两倍长度的数组：因为我们要让后面的数组能够与前面的数字比较。
我们可以使用取模的方法来实现，同样需要从两倍数组的长度遍历，对其取模才能得到对应值的下标

        stack<int> s;
        int n = nums.size();
        for (int i = 2*n-1; i >= 0; i--) {
            while (!s.empty() && nums[s.top()%n] <= nums[i%n]) {
                s.pop();
            }
            ans[i%n] = s.empty() ? -1 : nums[s.top()%n];
            s.push(i);
        }