LeetCode 第58天 | 739. 每日温度 496. 下一个更大元素 I 单调栈

739. 每日温度

对于单调栈的应用情况是求右边（左边）第一个大于或者小于该元素的值或者角标。右边大于则栈顶到栈底单调递增，右边小于则栈顶到栈底单调递减。具体方法是第一个元素直接入栈，遍历从1的元素开始，如果当前值小于等于栈顶元素，直接入栈；否则进行一个循环遍历，对于栈顶每一个小于当前元素，计算距离，并且出栈，直到全部满足要求出栈，再把当前值入栈。返回结果。

class Solution {
public:
    vector<int> dailyTemperatures(vector<int>& temperatures) {
        // vector<int> res(temperatures.size(), 0);
        // for (int i = 0; i < temperatures.size(); i++) {
        //     for (int j = i+1; j < temperatures.size(); j++) {
        //         if (temperatures[j] > temperatures[i]) {
        //             res[i] = j-i;
        //             break;
        //         }
        //     }
        // } 
        // return res;
        stack<int> st;
        vector<int> res(temperatures.size(), 0);
        // 第一个元素无脑入栈
        st.push(0);
        for (int i = 1; i < temperatures.size(); i++) {
            // res 存储的是角标
            if (temperatures[i] > temperatures[st.top()]) {
                // 如果当前值大于栈顶元素，一直出栈直到栈空或者不大于栈顶
                while(!st.empty() && temperatures[i] > temperatures[st.top()]){
                    res[st.top()] = i-st.top();
                    st.pop();
                }
                // 入栈
                st.push(i);
            }
            // 小于等于栈顶元素直接入栈
            else {
                st.push(i);
            }
        }
        return res;
    }
};

496. 下一个更大元素 I

这题题目太绕了，但是意思就是一组元素在另一组元素里，要保留前者的顺序，还要找到前者在后者中的最近较大元素。采用哈希和单调栈实现，其实暴力实现应该也能通过，但是能用好方法当然就用，就算记忆也是好的。mad算法题刷不动了，一边刷一边不会，老子要开始背了。

class Solution {
public:
    vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {
        // 哈希map加单调栈，单调栈就是右边第一个大于左边或者右边第一个小于左边
        stack<int> st;
        vector<int> res(nums1.size(), -1);
        st.push(0);

        unordered_map<int ,int> umap;
        for (int i = 0; i < nums1.size(); i++) {
            umap[nums1[i]] = i;
        }
        for (int i = 0; i < nums2.size(); i++) {
            if (nums2[i] <= nums2[st.top()]) {
                st.push(i);
            }
            else {
                while (!st.empty() && nums2[i] > nums2[st.top()]) {
                    if (umap.find(nums2[st.top()]) != umap.end()) {
                        int index = umap[nums2[st.top()]];
                        res[index] = nums2[i];
                    }
                    st.pop();
                }
                st.push(i);
            }
        }
        return res;
    }
};