代码随想录训练营第五十八天|739.每日温度、496.下一个更大元素

739.每日温度

链接：LeetCode739.每日温度

暴力解法—双层for循环

class Solution {
public:
    vector<int> dailyTemperatures(vector<int>& temperatures) {
        vector<int> ans(temperatures.size(),-1);
        for(int i=0;i<temperatures.size();++i){
            for(int j=i+1;j<temperatures.size();++j){
                if(temperatures[j]>temperatures[i]) {ans[i] = j-i;break;}
            }
            if(ans[i]==-1) ans[i]=0;
        }
        return ans;
    }
};

该解法会超时。时间复杂度为O(n^2)，空间复杂度为O(n);

单调栈

解题思路：需要一个辅助栈，存储温度以及对应的下标
由于要求的是比当天温度高的天数，所以在存储温度的同时还要存储温度所对应的下标也就是所谓的天数；由于要求的是更高的温度，于是当栈空或者当前温度小于等于栈顶温度时，该温度以及对应的下标进栈；反之，将栈顶元素出栈直到当前温度小于等于栈顶温度(这里使用的while循环)。之后将当前温度以及对应的下标进栈。每次出栈时要进行的操作：当前温度对应的下标-栈顶温度对应的下标就是ans[st.top().second]的值。

class Solution {
public:
    vector<int> dailyTemperatures(vector<int>& temperatures) {
        stack<pair<int,int>> st;//存储温度到天之间的映射
        st.push(pair<int,int>{temperatures[0],0});
        vector<int> ans(temperatures.size(),0);
        for(int i=1;i<temperatures.size();++i){
            while(!st.empty()&&temperatures[i]> st.top().first) {
                ans[st.top().second]=i-st.top().second;
                st.pop();
            }
            st.push(pair<int,int>{temperatures[i],i});
        }
        return ans;
    }
};

496.下一个更大元素

链接：LeetCode496.下一个更大元素
首先将nums1映射到哈希表中，遍历nums2的元素，依次入栈。栈空或者当前元素小于等于栈顶元素入栈；反之出栈直到当前元素小于等于栈顶元素，之后入栈当前元素。

class Solution {
public:
    vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {
        //将nums1存储到哈希表中
        unordered_map<int,int>ma;
        for(int i=0;i<nums1.size();++i) ma[nums1[i]] =i;
        vector<int> ans(nums1.size(),-1);
        stack<int> st;//单调栈，存储nums2的元素
        for(int i=0;i<nums2.size();++i){
            while(!st.empty()&&nums2[i]>st.top()){
                auto it = ma.find(st.top());
                if(it!=ma.end()) ans[it->second] = nums2[i];
                st.pop();
            }st.push(nums2[i]);
        }
        return ans;
    }
};