代码随想录算法训练营day58|| 第九章 单调栈

739. 每日温度

给定一个整数数组 temperatures ,表示每天的温度,返回一个数组 answer ,其中 answer[i] 是指对于第 i 天,下一个更高温度出现在几天后。如果气温在这之后都不会升高,请在该位置用 0 来代替。题目

首先想到用两层for循环遍历数组即可得到右边第一个比自己大的数,但是最后结果超时

class Solution {
public:
    vector dailyTemperatures(vector& temperatures) {
        vector answer(temperatures.size(),0);
        for(int i=0;itemperatures[i]){
                    answer[i]=j-i;
                    break;
                }
            }
        }
        return answer;
    }
};

单调栈的方法,单调栈常用于一维数组,求左边或者右边第一个比自己大或者比自己小的元素的位置,

class Solution {
public:
    vector dailyTemperatures(vector& temperatures) {
        vector answer(temperatures.size(),0);
        stack st;
        st.push(0);
        for(int i=1;itemperatures[st.top()]){
                    answer[st.top()]=i-st.top();
                    st.pop();
                }
                st.push(i);
            }
        }
        return answer;
    }
};

496.下一个更大元素 I

nums1 中数字 x 的 下一个更大元素 是指 x 在 nums2 中对应位置 右侧 的 第一个 比 x 大的元素。

给你两个 没有重复元素 的数组 nums1 和 nums2 ,下标从 0 开始计数,其中nums1 是 nums2 的子集。对于每个 0 <= i < nums1.length ,找出满足 nums1[i] == nums2[j] 的下标 j ,并且在 nums2 确定 nums2[j] 的 下一个更大元素 。如果不存在下一个更大元素,那么本次查询的答案是 -1 。

返回一个长度为 nums1.length 的数组 ans 作为答案,满足 ans[i] 是如上所述的 下一个更大元素 。题目

class Solution {
public:
    vector nextGreaterElement(vector& nums1, vector& nums2) {
        vector result(nums1.size(),-1);
        if(nums1.size()==0){
            return result;
        }
        stack st;
        unordered_map mp;
        for(int i=0;inums2[st.top()]){
                    if(mp.count(nums2[st.top()])>0){
                        int index=mp[nums2[st.top()]];
                        result[index]=nums2[i];
                    }
                    st.pop();
                }
                st.push(i);
            }
        }
        return result;
    }
};

你可能感兴趣的:(算法,leetcode)