性质
单调栈是一种特殊的栈,分为单调递减栈和单调递增栈,站内的元素都保持单调性
单调递减栈
栈底到栈顶是一个递减的数列,遍历一个数组当遍历到的数字大于栈顶时(一般就是题目的答案)跳过如果小于栈顶就插入
例题(496. 下一个更大元素 I(力扣))
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 。
示例 2:
输入:nums1 = [2,4], nums2 = [1,2,3,4].
输出:[3,-1]
解释:nums1 中每个值的下一个更大元素如下所述:
- 2 ,用加粗斜体标识,nums2 = [1,2,3,4]。下一个更大元素是 3 。
- 4 ,用加粗斜体标识,nums2 = [1,2,3,4]。不存在下一个更大元素,所以答案是 -1 。
提示:
1 <= nums1.length <= nums2.length <= 1000
0 <= nums1[i], nums2[i] <= 104
nums1和nums2中所有整数 互不相同
nums1 中的所有整数同样出现在 nums2 中
暴力写法
class Solution {
public:
vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {
vector<int>ans;
int c=0;
for(int i=0;i<nums1.size();i++){
int k=0;
for(int j=0;j<nums2.size();j++){
if(nums1[i]==nums2[j])k=1;
if(k==1&&nums2[j]>nums1[i]){
ans.push_back(nums2[j]);
break;
}
if(j==nums2.size()-1)ans.push_back(-1);
}
}
return ans;
}
};
单调栈+哈希
解题思路
class Solution {
public:
vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {
stack<int>a; //建一个栈用来做单调栈
map<int,int>s; //用记录nums1中每个元素在nums2中所求的后一个大的值或者-1;
for(int i=0;i<nums2.size();i++){
if(a.empty()){
a.push(nums2[i]);
continue;
}
if(nums2[i]>a.top()){ //如果遍历到的数大于栈顶元素的话,记录该值并把栈顶弹出
s[a.top()]=nums2[i];
a.pop();
while(!a.empty()){ //与第一个值比后,继续与后面的比较,到栈空或者小于栈
//顶元素后停止并插入该元素
if(nums2[i]>a.top()){
s[a.top()]=nums2[i];
a.pop();
}else{
break;
}
}
a.push(nums2[i]);
}else{
a.push(nums2[i]);
}
}
vector<int>ans; //遍历并记录答案
for(int i=0;i<nums1.size();i++){
if(s[nums1[i]]==0)s[nums1[i]]=-1;
ans.push_back(s[nums1[i]]);
}
return ans;
}
};