算法题打卡day58-单调栈 | 739. 每日温度、496.下一个更大元素I

739. 每日温度 - 力扣（LeetCode）

状态：暴力超时，查看思路后AC。

利用持续维护一个单调递增的栈来实现一次遍历就可以得到答案，代码如下：

class Solution {
public:
    vector<int> dailyTemperatures(vector<int>& temperatures) {
        int len = temperatures.size();
        vector<int> res(len, 0);
        stack<int> st;
        for(int i = 0; i < len; ++i){
            while(!st.empty() && temperatures[st.top()] < temperatures[i]){
                res[st.top()] = i - st.top();
                st.pop();
            }
            st.push(i);
        }
        return res;
    }
};

496. 下一个更大元素 I - 力扣（LeetCode）

状态：查看思路后AC。

上一题的变体，主要是添加了一个unordered_map来对存在的元素进行匹配，代码如下：

class Solution {
public:
    vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {
        int len1 = nums1.size(), len2 = nums2.size();
        vector<int> res(len1, -1);
        unordered_map<int, int> unmap;
        for(int i = 0; i < len1; ++i){
            // key 是数组1的值，value是索引
            unmap[nums1[i]] = i;
        }
 
        stack<int> st;
        for(int j = 0; j < len2; ++j){
            while(!st.empty() && nums2[st.top()] < nums2[j]){
                if(unmap.count(nums2[st.top()]) > 0){
                    // nums1中存在
                    int index1 = unmap[nums2[st.top()]];
                    res[index1] = nums2[j];
                }
                st.pop();
            }
            st.push(j);
        }
        return res;
    }
};