【Day62】代码随想录之单调栈_739. 每日温度_496.下一个更大元素 I

文章目录

1. 739. 每日温度

分析：
找下一个更高温度出现在几天后，即当前位置右侧出现的一个比它更大的值，如果是暴力搜索，两层for，时间复杂度O(n)， 数据范围10,时间最大是10，这可能会超，我试了试，确实超了，用一个很大的全都一样的数组就超了。

单调栈是栈里是数组的下标，下标对应的数组元素是有序的。这里使用从栈顶到栈底递增的方法：这样遇到比栈顶更大的就可以根据下标求出距离了。遇到的元素和栈顶的元素会有三种情况：
栈顶元素大于遍历元素：入栈保留。
栈顶元素等于遍历元素：入栈保留。
栈顶元素小于遍历元素：弹出栈顶，下标相减再减一求出距离。根据栈里的单调，当前遍历元素和栈顶一直抵消，直到栈顶元素大于遍历元素。

代码：

class Solution {
public:
    vector<int> dailyTemperatures(vector<int>& temperatures) {
        vector<int> answer(temperatures.size(), 0);
        stack<int> st;//递增栈，存放的是下标
        st.push(0);

        for(int i = 1; i < temperatures.size(); i++){
            if(temperatures[i] < temperatures[st.top()]){//小于栈顶
                st.push(i);
            }
            else if(temperatures[i] == temperatures[st.top()]){
                st.push(i);
            }
            else{
                while(!st.empty() && temperatures[i] > temperatures[st.top()]){
                    answer[st.top()] = i - st.top();
                    st.pop();
                }
                st.push(i);
            }        
        }

        if(!st.empty()){
            answer[st.top()] = 0;
            st.pop();
        }

        return answer;
    }
};

2. 496.下一个更大元素 I

分析：
和温度一样要找出右侧第一个比当前元素大的值，但是在找的时候需要查询当前元素在nums1中在不在，在的话保存值，而不是距离。因为用到查询是不是在nums1中存在，unordered_map<int, int>哈希表的底层逻辑查找效率是O(1)，保存nums1中的数组下标和数组元素。

代码：

class Solution {
public:
    vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {
        stack<int> st;
        vector<int> answer(nums1.size(), -1);
        if(nums1.size() == 0) return answer;

        unordered_map<int, int> umap;
        for(int i = 0; i < nums1.size(); i++){
            umap[nums1[i]] = i;//保存值和下标，下标用来更新result
        }
        st.push(0);
        for(int i = 1; i < nums2.size(); i++){
            if(nums2[i] < nums2[st.top()]){
                st.push(i);
            }
            else if(nums2[i] == nums2[st.top()]){
                st.push(i);
            }
            else{
                while(!st.empty() && nums2[i] > nums2[st.top()]){
                    if(umap.count(nums2[st.top()]) > 0) answer[umap[nums2[st.top()]]] = nums2[i];
                    st.pop();
                }
                st.push(i);
            }
        }
        
        return answer;
        
    }
};