【c++刷题笔记-单调栈】day48： 739. 每日温度 、496.下一个更大元素 I 、503.下一个更大元素II

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;
    }
};

总结

单调栈，单调递增是求下一个更大的元素，单调递减是下一个更小的元素。单调栈就是记录之前遍历过的元素。