739. 每日温度 - 力扣(LeetCode)
思路:使用单调栈遍历,循环找栈中比栈顶数小的就出栈,并且记录值。比栈顶数大的就入栈,形成单调递增的栈
class Solution {
public:
vector<int> dailyTemperatures(vector<int>& temperatures) {
int n=temperatures.size();
stack<int>st;
vector<int>ans(n,0);
for(int i=0;i<n;i++){
//循环找栈中比栈顶数小的就出栈,并且记录值。比栈顶数大的就入栈,形成单调递增的栈
while(!st.empty()&&temperatures[i]>temperatures[st.top()]){
ans[st.top()]=i-st.top();
st.pop();
}
st.push(i);
}
return ans;
}
};
496. 下一个更大元素 I - 力扣(LeetCode)
思路:同上,但需要给num1做哈希映射,方便找出num2的中是否存在对应的值
class Solution {
public:
vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {
vector<int>ans(nums1.size(),-1);
stack<int>st;
if(nums1.size()==0){
return ans;
}
unordered_map<int,int>map;
for(int i=0;i<nums1.size();i++){
map[nums1[i]]=i;//映射nums1的下标
}
st.push(0);
for(int i=1;i<nums2.size();i++){
while(!st.empty()&&nums2[i]>nums2[st.top()]){
if(map.count(nums2[st.top()])>0){//找到比下一个更大的元素,判断是否在nums1中
int index=map[nums2[st.top()]];
ans[index]=nums2[i];
}
st.pop();
}
st.push(i);
}
return ans;
}
};
503. 下一个更大元素 II - 力扣(LeetCode)
思路:同上,需要使用取模运算模拟复制一个相同的数组拼接在末尾。
重点:头尾相连和循环问题都可以用此方法,注意入栈下标不能大于数组本身的大小
class Solution {
public:
vector<int> nextGreaterElements(vector<int>& nums) {
int n=nums.size();
vector<int>ans(n,-1);
stack<int>st;
for(int i=0;i<2*n-1;i++){//取模模拟拼接数组
int x=nums[i%n];
//找到下一个更大的元素
while(!st.empty()&&x>nums[st.top()]){
ans[st.top()]=x;
st.pop();
}
//下标不能大于nums.size()
if(i<n){
st.push(i);
}
}
return ans;
}
};
总结
单调栈,单调递增是求下一个更大的元素,单调递减是下一个更小的元素。单调栈就是记录之前遍历过的元素。