day-58 代码随想录算法训练营（19）单调栈 part 01

739.每日温度

思路一：双指针暴力超时

思路二：单调栈

class Solution {
public:
    vector<int> dailyTemperatures(vector<int>& temperatures) {
        int n=temperatures.size();
        stack<int>st;
        vector<int>res(n,0);
        for(int i=0;i<n;i++){
            //数组不为空并且，当前值比栈顶元素大时
            while(!st.empty() && temperatures[i]>temperatures[st.top()]){
                res[st.top()]=i-st.top();//记录中间隔了几个
                st.pop();//记录完弹出
            }
            //cout<<st.top();
            st.push(i);//压入下标
        }
        return res;
    }
};

496.下一个更大元素 I

思路一：单调栈+unordered_map 进行查找

• 先求出nums2中所有元素，右侧第一个比它大的值；加入中间数组
• 然后对nums2创建 元素值：下标 的键值对关系（下标从0开始，这里作+1处理）
• 最后在nums1中直接通过 unordered_map 进行查找下标（需要 -1 处理），得到下标后从中间数组中获取结果

class Solution {
public:
    vector<int> judge(vector<int>&ans){//一摸一样
        int n=ans.size();
        stack<int>st;
        vector<int>mid(n,-1);
        for(int i=0;i<n;i++){
            while(!st.empty() && ans[i]>ans[st.top()]){
                mid[st.top()]=ans[i];
                st.pop();
            }
            st.push(i);
        }
        return mid;
    }
    vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {
        int n=nums1.size(),m=nums2.size();
        unordered_map<int,int>map;
        for(int i=0;i<m;i++) map[nums2[i]]=i+1;//记录nums2数组中 值：下标
        vector<int>res(n,-1);
        vector<int>mid=judge(nums2);//得到nums2中，每个元素右侧第一个比它大的值
        for(int i=0;i<n;i++) res[i]=mid[map[nums1[i]]-1];//通过map的值找到下标，在mid中找到结果
        return res;
    }
};