单调栈学习思路

单调栈:
构造一个单调递增或递减的栈。

作用范围:
找左侧或右侧比当前元素大一个或小一个的元素或者计算两个元素之间的距离

过程:
举例:找当前元素右侧比当前元素大的第一个元素,第一个元素直接入栈,对后续元素逐个判断,比栈顶元素小直接入栈,
比栈顶元素大则用当前元素的位置减去栈顶元素的位置,得到栈顶元素到比当前元素大的第一个元素的距离。
并将栈顶元素出栈,继续比较直到栈顶元素小于当前元素后,把当前元素入栈。

原理:
递增单调栈维持了栈内元素单调递增的顺序,并通过比较的方法将比栈顶元素小的元素入栈,把比栈顶元素的大的元素与
栈顶元素计算位置并出栈,直至栈内元素都比栈顶元素大或者为空,确保加入当前元素依旧满足单调栈的性质。
这样通过遍历可以找到每个元素右侧第一个比它大的值,因为栈顶元素永远是栈内最小的,遍历的元素比栈顶元素大则可以
逐个比较确保找到每个元素右侧第一个比它大的值。

通俗易懂的解释:
找第一个更大的元素:遍历时与栈顶元素比较,小于栈顶进栈,大于栈顶元素时与栈内元素比较,比当前元素小的栈内元素逐个弹出(此时当前元素为栈内元素的第一个更大的元素)这样在查找的同时能维持栈内顺序。

例题:下一个更大的元素
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 。

class Solution {
public:
vector nextGreaterElement(vector& nums1, vector& nums2) {
unordered_map<int,int>map;

    for(int i = 0;i<nums1.size();++i){
        map[nums1[i]] = i+1;
    }
    
    vector<int> res(nums1.size(),-1);
    
    stack<int> s;
    
    for(int i = 0;i<nums2.size();++i){
        
        while(!s.empty() && nums2[i] > nums2[s.top()]){
            
            cout <<"map = " << map[nums2[s.top()]] << endl;
            
            if(map[nums2[s.top()]]){
                
                res[map[nums2[s.top()]]-1] = nums2[i];
                
            }
            
            s.pop();
        }
        s.push(i);
    }
    return res;
}

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

示例1:

a0b5725179e8d2e8540134afe72860be

输入: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& height) {
int res = 0;
stack s;
//通过栈不断的找凹槽,若不存在凹槽则弹出
//单调栈的思路就是找右侧第一个更大的柱子
//找到右边第一个更大的柱子后,找当前元素左侧第一个更大的柱子
//这样就能找到所有的凹槽 计算出凹槽内的雨水量
for(int i = 0;i<height.size();++i){
while(!s.empty() && height[i] >= height[s.top()]){
int temp = height[s.top()];
s.pop();
if(!s.empty()){
int h = min(height[i],height[s.top()])-temp;
int w = i - s.top() - 1;
res += (h * w);
}
}
s.push(i);
}
return res;
}
};

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值