第53天-DP-第十章 单调栈 ● 739. 每日温度 ● 496.下一个更大元素 I

1. 每日温度]

- 代码随想录链接

class Solution {
public:
    vector<int> dailyTemperatures(vector<int>& temperatures) {
        // 1.栈中存放的是温度列表中的下标元素i
        stack<int> st;
        vector<int> result(temperatures.size(), 0);
        // st.push(0);

        // // 2.分三种情况，遍历的值大于、等于、小于栈顶的值
        // 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()]){         // 还需要判断stack是否为空
        //             result[st.top()] = i - st.top();
        //             st.pop();
        //         }
        //         st.push(i);
        //     }
        // }

        // 精简版
        for(int i=0; i<temperatures.size(); i++){
            while(!st.empty() && temperatures[i] > temperatures[st.top()]){         // 还需要判断stack是否为空
                result[st.top()] = i - st.top();
                st.pop();
            }
            st.push(i);
        }

        return result;
    }
};

2. 下一个更大元素

- 代码随想录链接

class Solution {
public:


    vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {

        // s存储的是 nums2[j]中对应的下标
        stack<int> st;
        vector<int> result(nums1.size(), -1);
        // 使用一个map存储 nums1中的元素和下边，在遍历nums2的时候判断元素时候nums1中出现过
        unordered_map<int, int> umap;
        for(int i=0; i<nums1.size(); i++){
            umap[nums1[i]] = i;
        }

        st.push(0);

        for(int j=1; j<nums2.size(); j++){
            if(nums2[j] < nums2[st.top()]){
                st.push(j);
            }else if(nums2[j] == nums2[st.top()]){      // 因为没有重复元素，所以这种情况可以不用考虑
                st.push(j);
            }else{
                while(!st.empty() && nums2[j] > nums2[st.top()]){
                    if(umap.find(nums2[st.top()]) != umap.end()){
                        int index = umap[nums2[st.top()]];
                        result[index] = nums2[j];
                    }
                    st.pop();
                }
            st.push(j);
            }
        }

        return result;
    }
};