单调栈模板

分两种情况,找左边和找右边,找左边从前到后入栈,找右边从后到前入栈

数组模拟:

//常见模型:找出每个数左边离它最近的比它大/小的数

```cpp
int tt = 0;
for (int i = 1; i <= n; i ++ )
{
    while (tt && check(stk[tt], i)) tt -- ;
    stk[ ++ tt] = i;
}

STL模板

```cpp
stack<int>s;
for(int i=0;i<n;i++){
while(!s.empty()&&check(num[i],s.top())) s.pop();//检查栈顶元素是不是满足要求
//做题目要求的任务
s.push(i);//将数字加入栈中
}

例题:力扣496
给你两个 没有重复元素 的数组 nums1 和 nums2 ,其中nums1 是 nums2 的子集。
请你找出 nums1 中每个元素在 nums2 中的下一个比其大的值。
nums1 中数字 x 的下一个更大元素是指 x 在 nums2 中对应位置的右边的第一个比 x 大的元素。如果不存在,对应位置输出 -1 。

//这是个典型的从右到左找套用模板即可
vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {
 stack<int>s;
 vector<int>v;
 map<int,int>mp;
 for(int i=nums2.size()-1;i>=0;i--){
     while(!s.empty()&&s.top()<nums2[i]) s.pop();//第一步
     mp[nums2[i]]=s.empty()?-1:s.top();//第二步
     s.push(nums2[i]);//第三步
 }
 for(auto it:nums1)
     v.push_back(mp[it]);
 return v;
 }
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
单调栈是一种特殊的栈结构,它可以快速找到栈中的最大值或最小值。单调栈常用于解决一些数组、字符串中的问题,例如求下一个更大的元素、求前一个更大的元素等等。 下面是单调栈的代码模板: ``` stack<int> mono_stack; for (int i = 0; i < n; i++) { while (!mono_stack.empty() && mono_stack.top() > x[i]) { mono_stack.pop(); } mono_stack.push(x[i]); } ``` 其中,`x` 是一个数组,`n` 是数组的长度,`mono_stack` 是单调栈。这段代码的意思是:遍历数组 `x`,对于每个元素 `x[i]`,将其压入栈中,然后不断弹出栈顶元素,直到栈为空或者栈顶元素不大于 `x[i]`。这样,最终单调栈中的元素就是一个单调递增的序列。 如果要求下一个更大的元素,可以在遍历数组时,记录每个元素的下标,然后在弹出栈顶元素时,更新下标对应的答案。代码如下: ``` stack<pair<int, int>> mono_stack; for (int i = 0; i < n; i++) { while (!mono_stack.empty() && mono_stack.top().first < x[i]) { ans[mono_stack.top().second] = x[i]; mono_stack.pop(); } mono_stack.push(make_pair(x[i], i)); } ``` 其中,`ans` 是答案数组,`ans[i]` 表示第 `i` 个元素的下一个更大的元素。在弹出栈顶元素时,更新对应下标的答案即可。 如果要求前一个更大的元素,可以将数组逆序遍历,然后按照同样的方式求解。代码如下: ``` stack<pair<int, int>> mono_stack; for (int i = n - 1; i >= 0; i--) { while (!mono_stack.empty() && mono_stack.top().first < x[i]) { ans[mono_stack.top().second] = x[i]; mono_stack.pop(); } mono_stack.push(make_pair(x[i], i)); } ``` 以上就是单调栈的代码模板,可以根据具体问题进行修改和扩展。

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值