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;}
};
总结
单调栈,单调递增是求下一个更大的元素,单调递减是下一个更小的元素。单调栈就是记录之前遍历过的元素。