Day58| 739 每日温度 496 下一个更大元素 I

目录

739 每日温度  

496 下一个更大元素 I   

---

739 每日温度  

class Solution {
public:
    vector<int> dailyTemperatures(vector<int>& temperatures) {
        //栈里面放的是下标
        stack<int> st;
        //result放的是差值也就是结果
        vector<int> result(temperatures.size(), 0);

        st.push(0);
        
        for(int i = 1; i < temperatures.size(); i++){
            //1 栈元素比当前元素大,那我们就要把元素插进去
            if(temperatures[st.top()] > temperatures[i]){
                st.push(i);
            }//2 栈元素等于当前元素,也要插进去
            else if(temperatures[st.top()] == temperatures[i]){
                st.push(i);
            }//3 栈元素比当前元素小,说明我们找到要的元素了,就可以开始操作了
            else{
                //while是因为当前的元素可能比栈里面的好几个元素都大，那就说明这几个元素都找到了
                while(!st.empty() && temperatures[st.top()] < temperatures[i]){
                    result[st.top()] = i - st.top();//赋值
                    st.pop();//把插进去的栈顶元素pop掉
                }
                //别忘记了把当前元素插进来
                st.push(i);
            }
        }
        return result;
    }
};

496 下一个更大元素 I   

class Solution {
public:
    vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {
        stack<int> st;
        vector<int> result(nums1.size(), -1);
        if (nums1.size() == 0) return result;
        unordered_map<int, int> unmp;
        int index = 0;//记录nums2在nums1里面出现的位置

        //把nums1里面的元素插进map做个记录,记一下对应元素出现的位置(下标),方便result进行插入
        for(int i =0; i < nums1.size(); i++){
            unmp[nums1[i]] = i;
        }
        //用单调栈的方法来找nums2里面元素的下一个更大元素
        st.push(0);

        for(int i = 1; i < nums2.size(); i++){
            if(nums2[i] < nums2[st.top()]){
                st.push(i);
            }else if(nums2[i] < nums2[st.top()]){
                st.push(i);
            }else{
                //如果栈不空且当前值比栈顶值大，我们就找到了
                while(!st.empty() && nums2[i] > nums2[st.top()]){
                    //在map里面查询看看nums1里面是否存在当前nums2的值
                    if(unmp.count(nums2[st.top()]) > 0){
                        //如果存在，就算一下当前值在nums1的位置然后在result的对应位置插进去
                        index = unmp[nums2[st.top()]];
                        result[index] = nums2[i];
                    }
                    st.pop();
                }
                st.push(i);
            }
        }
        return result;

   
    }
};