代码随想录算法训练营第四十八天|LeetCode739. 每日温度,LeetCode496. 下一个更大元素 I,LeetCode503. 下一个更大元素 II

LeetCode739. 每日温度

题目链接:739. 每日温度 - 力扣(LeetCode)
文章讲解:代码随想录
视频讲解:单调栈,你该了解的,这里都讲了!LeetCode:739.每日温度_哔哩哔哩_bilibili

思路:这道题最简单也最直接的思路是用双指针,也就是两层for循环,不过这样会超时。于是我们想到了单调栈的思路。单调栈求解是这样的:首先把第一个数的下标放进去,然后放第二个数,如果小于等于第一个数,那么直接push进去,如果大于,那么直接while循环把所有小于这个数的头pop出栈,并且保存两个下标之差进result数组。直到不大于栈顶的元素,就push进去。

代码:

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

LeetCode496. 下一个更大元素 I

题目链接:496. 下一个更大元素 I - 力扣(LeetCode)
文章讲解:代码随想录
视频讲解:单调栈,套上一个壳子就有点绕了| LeetCode:496.下一个更大元素_哔哩哔哩_bilibili

思路:和上一题思路有点像,但是很绕。首先我们要用unordered_map来存储nums1中的元素和下标的对应关系,然后使用单调栈遍历nums2,然后当遇到当前元素比栈顶元素大的情况时,我们去umap中找到这个数和这个数的下标,把result[这个数的下标]保存为nums[i]。

代码:

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

LeetCode503. 下一个更大元素 II

题目链接:503. 下一个更大元素 II - 力扣(LeetCode)
文章讲解:代码随想录
视频讲解:单调栈,成环了可怎么办?LeetCode:503.下一个更大元素II_哔哩哔哩_bilibili

思路:和每日温度那一道题的代码很像,只不过需要把所有i替换成i%nums.size(),这样做相当于在原来那个的后面又接上了一段一样的数组,这样就可以起到循环的效果。

代码:

class Solution {
public:
    vector<int> nextGreaterElements(vector<int>& nums) {
        vector<int>result(nums.size(),-1);
        if(nums.empty()){
            return result;
        }
        stack<int>st;
        st.push(0);
        for(int i=1;i<nums.size()*2;i++){
            if(nums[i%nums.size()]<=nums[st.top()]){
                st.push(i%nums.size());
            }
            else{
                while(!st.empty()&&nums[i%nums.size()]>nums[st.top()]){
                    result[st.top()]=nums[i%nums.size()];
                    st.pop();
                }
                st.push(i%nums.size());
            }
        }
        return result;
    }
};
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值