第五十七天|739. 每日温度 496. 下一个更大元素 I

739. 每日温度

单调栈就是把数值一个个放进栈中，每次放之前进行对比，如果要进来的数值比栈顶的数值大/小就把栈顶弹出，使栈顶到栈底的数值一定是递增/递减的。

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

 496. 下一个更大元素 I

单调栈加哈希

class Solution {
public:
    vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {
      if(nums1.size()==0)return {};
      vector<int> res(nums1.size(),-1);
      stack<int> st;
      unordered_map<int,int> hash;
      for(int i=0;i<nums1.size();i++){
        hash[nums1[i]]=i;
      }
      for(int i=0;i<nums2.size();i++){
        while(!st.empty()&&nums2[i]>nums2[st.top()]){
          int index=hash[nums2[st.top()]];
          res[index]=nums2[i];
          st.pop();
        }
        if(hash.find(nums2[i])!=hash.end())st.push(i);
      }

      return res;
    }
};