单调栈篇--代码随想录算法训练营第三十天|单调栈介绍，739. 每日温度，496.下一个更大元素 I，503.下一个更大元素II，42. 接雨水，84.柱状图中最大的矩形

热爱编程的OP

已于 2024-08-15 15:51:03 修改

阅读量748

点赞数 17

分类专栏： leetcode 文章标签：算法 leetcode 学习 c++

于 2024-08-15 11:42:53 首次发布

本文链接：https://blog.csdn.net/m0_67804957/article/details/141203442

版权

leetcode 专栏收录该内容

25 篇文章 0 订阅

订阅专栏

单调栈介绍

使用场景：

求某个元素的左边/右边第一个比其大的数

对该方法理解：

单调栈的本质是空间换时间，用一个栈来记录我们遍历过的元素。如果求一个元素右边第一个更大元素，单调栈就是递增的，如果求一个元素右边第一个更小元素，单调栈就是递减的。

三个判断条件：

当前遍历的元素T[i]小于栈顶元素T[st.top()]的情况
当前遍历的元素T[i]等于栈顶元素T[st.top()]的情况
当前遍历的元素T[i]大于栈顶元素T[st.top()]的情况

模板：


    tpye function(vector<int>& nums) {
        stack<int> st; //定义单调栈
        st.push(0); //将首元素位置下标入栈
        type result; //根据题目需求定义结果变量result
        for(int i = 1; i < nums.size(); i++)
        {
            while(!st.empty() && nums[i] > nums[st.top()])
            {
               使用result保存运算结果; 
               st.pop();
            }
            st.push(i);
        }
        return result;
    }

739. 每日温度

讲解视频：

单调栈，你该了解的，这里都讲了！LeetCode:739.每日温度

题目描述：

给定一个整数数组 temperatures ，表示每天的温度，返回一个数组 answer ，其中 answer[i] 是指对于第 i 天，下一个更高温度出现在几天后。如果气温在这之后都不会升高，请在该位置用 0 来代替。

示例 1:

输入: temperatures = [73,74,75,71,69,72,76,73]
输出: [1,1,4,2,1,1,0,0]

解题思路：

使用递增单调栈

代码：

class Solution {
public:
    vector<int> dailyTemperatures(vector<int>& temperatures) {
        stack<int> s;
        s.push(0);
        vector<int> result(temperatures.size());
        for(int i = 1; i < temperatures.size(); i++)
        {
            while(!s.empty() && temperatures[i] > temperatures[s.top()])
            {
               result[s.top()] = i - s.top();
                s.pop();
            }
            s.push(i);
        }
        return result;
    }
};

496.下一个更大元素 I

讲解视频：

单调栈，套上一个壳子就有点绕了| LeetCode:496.下一个更大元素

题目描述：

nums1 中数字 x 的 下一个更大元素 是指 x 在 nums2 中对应位置右侧的 第一个 比 x 大的元素。给你两个 没有重复元素 的数组 nums1 和 nums2 ，下标从 0 开始计数，其中nums1 是 nums2 的子集。对于每个 0 <= i < nums1.length ，找出满足 nums1[i] == nums2[j] 的下标 j ，并且在 nums2 确定 nums2[j] 的 下一个更大元素 。如果不存在下一个更大元素，那么本次查询的答案是 -1 。返回一个长度为 nums1.length 的数组 ans 作为答案，满足 ans[i] 是如上所述的 下一个更大元素 。

示例 1：

输入：nums1 = [4,1,2], nums2 = [1,3,4,2].
输出：[-1,3,-1]
解释：nums1 中每个值的下一个更大元素如下所述：
- 4 ，用加粗斜体标识，nums2 = [1,3,4,2]。不存在下一个更大元素，所以答案是 -1 。
- 1 ，用加粗斜体标识，nums2 = [1,3,4,2]。下一个更大元素是 3 。
- 2 ，用加粗斜体标识，nums2 = [1,3,4,2]。不存在下一个更大元素，所以答案是 -1 。

解题思路：

使用递增单调栈；
使用哈希表保存nums1数据位置，便于nums2查找。使用count / find方法查找。

代码：

class Solution {
public:
    vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {
        vector<int> result(nums1.size(),-1);
        stack<int> st;
        unordered_map<int,int> hash;
        for(int i = 0; i < nums1.size(); i++)
            hash[nums1[i]] = i;

        st.push(0);
        for(int i = 1; i < nums2.size(); i++)
        {
            while(!st.empty() && nums2[i] > nums2[st.top()])
            {
                if(hash.count(nums2[st.top()]) > 0)
                    result[hash[nums2[st.top()]]] = nums2[i];
                st.pop();
            }
            st.push(i);
        }
        
        return result;
    }
};

503.下一个更大元素II

题目链接：. - 力扣（LeetCode）

讲解视频：

单调栈，成环了可怎么办？LeetCode：503.下一个更大元素II

题目描述：

给定一个循环数组 nums （ nums[nums.length - 1] 的下一个元素是 nums[0] ），返回 nums 中每个元素的 下一个更大元素 。数字 x 的 下一个更大的元素 是按数组遍历顺序，这个数字之后的第一个比它更大的数，这意味着你应该循环地搜索它的下一个更大的数。如果不存在，则输出 -1 。

示例 1:

输入: nums = [1,2,1]
输出: [2,-1,2]
解释: 第一个 1 的下一个更大的数是 2；
数字 2 找不到下一个更大的数； 
第二个 1 的下一个最大的数需要循环搜索，结果也是 2。

解题思路：

使用递增单调栈

对于循环数组，有以下两种思路：

将两个nums数组拼接为一个nums数组，遍历一次
将一个nums数组遍历两遍，此时下标使用i%nums.size()

代码：

1、两个nums数组拼接

class Solution {
public:
    vector<int> nextGreaterElements(vector<int>& nums) {
        vector<int> cirNums(nums.begin(),nums.end());
        cirNums.insert(cirNums.end(),nums.begin(),nums.end());
        vector<int> result(cirNums.size(),-1);
        stack<int> st;
        st.push(0);
        for(int i = 1; i < cirNums.size(); i++)
        {
            while(!st.empty() && cirNums[i] > cirNums[st.top()])
            {
                result[st.top()] = cirNums[i];
                st.pop();
            }
            st.push(i);
        }
        result.resize(nums.size());
        return result;
    }
};

2、1个nums遍历两遍

class Solution {
public:
    vector<int> nextGreaterElements(vector<int>& nums) {
        vector<int> result(nums.size(),-1);
        stack<int> st;
        st.push(0);
        for(int i = 1; i < nums.size() * 2; i++)
        {
            while(!st.empty() && nums[i % nums.size()] > nums[st.top()])
            {
                result[st.top()] = nums[i % nums.size()];
                st.pop();
            }
            st.push(i % nums.size());
        }
        return result;
    }
};

42. 接雨水

题目链接：. - 力扣（LeetCode）

讲解视频：

单调栈，经典来袭！LeetCode:42.接雨水

题目描述：

给定 n 个非负整数表示每个宽度为 1 的柱子的高度图，计算按此排列的柱子，下雨之后能接多少雨水。

示例 1：

输入：height = [0,1,0,2,1,0,1,3,2,1,2,1]
输出：6
解释：上面是由数组 [0,1,0,2,1,0,1,3,2,1,2,1] 表示的高度图，在这种情况下，可以接 6 个单位的雨水（蓝色部分表示雨水）。

解题思路：

使用递增单调栈。

跟前几题不同处在于本题需要同时考虑三个元素，即栈顶和栈顶的下一个元素以及要入栈的元素。要保证可以接到雨水，则要满足左右两个元素比中间元素大，根据单调栈特征知栈顶的下一个元素一定比栈顶元素大，故只需要入栈的元素比栈顶元素大即可。

代码：

class Solution {
public:
    int trap(vector<int>& height) {
        if(height.size() == 1 || height.size() == 2) return 0;
        stack<int> st;
        st.push(0);
        int sum = 0;
        for(int i = 1; i < height.size(); i++)
        {
            while(!st.empty() && height[st.top()] < height[i])
            {
                int mid = st.top();
                st.pop();
                if(!st.empty())
                {
                    int top = st.top();
                    int h = min(height[top],height[i]) - height[mid];
                    sum += h * (i - top - 1);
                }
            }
            st.push(i);
        }
        return sum;
    }
};

84.柱状图中最大的矩形

题目链接：. - 力扣（LeetCode）

讲解视频：

单调栈，又一次经典来袭！ LeetCode：84.柱状图中最大的矩形

题目描述：

给定 n 个非负整数，用来表示柱状图中各个柱子的高度。每个柱子彼此相邻，且宽度为 1 。

求在该柱状图中，能够勾勒出来的矩形的最大面积。

示例 1:

输入：heights = [2,1,5,6,2,3]
输出：10
解释：最大的矩形为图中红色区域，面积为 10

解题思路：

使用递减单调栈。

同时考虑三个元素，即栈顶和栈顶的下一个元素以及要入栈的元素。与上一题接雨水题目不同点在于本题若最后只剩下两个元素，依旧可以有解，故首位各填充一个0来简化代码并避免剩下2个元素时无法计算的问题。

代码：

class Solution {
public:
    int largestRectangleArea(vector<int>& heights) {
        heights.insert(heights.begin(),0);
        heights.push_back(0);
        stack<int> st;
        st.push(0);
        int area = 0;
        for(int i = 1; i < heights.size(); i++)
        {
            while(!st.empty() && heights[st.top()] > heights[i])
            {
                int mid = st.top();
                st.pop();
                if(!st.empty())
                {
                    int w = i - st.top() - 1;
                    area = max(area, w * heights[mid]);
                }
            }
            st.push(i);
        }
        return area;
    }
};

热爱编程的OP

关注

17
点赞
踩
13

收藏

觉得还不错? 一键收藏
0
评论
单调栈篇--代码随想录算法训练营第三十天|单调栈介绍，739. 每日温度，496.下一个更大元素 I，503.下一个更大元素II，42. 接雨水，84.柱状图中最大的矩形

单调栈介绍，739. 每日温度，496.下一个更大元素 I，503.下一个更大元素II，42. 接雨水，84.柱状图中最大的矩形
复制链接

扫一扫