【Leetcode单调栈】496. 下一个更大元素 I(单调栈加一点点哈希!!和上一题的单调栈一摸一样!!)

Leetcode496

1.问题描述

在这里插入图片描述

2.解决方案(单调栈加哈希)


a.思路

1.这道题简直和上一题【Leetcode单调栈】739. 每日温度(有关于单调栈类型题的深入理解概括!!)一模一样啊,都是求数组中元素右边第一个比自己大的元素,这不是和上一题一摸一样嘛,上一题只不过是求所以元素,这一题求的是子集,反正我全求出来不就好了,子集那就从结果中找到子集就好了!
2.先遍历nums2利用单调栈把nums2中元素右边第一个比自己大的值求出来并放入map中
3.然后遍历nums1直接根据key取结果
4.也可以归纳为就是哈希表的建立问题,目前我们遇见过,先遍历一遍某个集合数组等,直接建立哈希表,这道题目是利用了单调栈的特性很符合题意,也是遍历一遍集合然后建立哈希表。

在这里插入图片描述
在这里插入图片描述



b.代码

1.本题的思路和【Leetcode单调栈】739. 每日温度(有关于单调栈类型题的深入理解概括!!)是一样的,但是代码的实现有所不同!
2.上一题遇到元素大于栈顶元素,是利用while一直出栈并记录结果,而本题是出一次栈,i–,使得下一次还是这个元素,这样怎么说更优雅一点好像,也还行
3.还有上一题是记录下标,本题是记录元素都是可以的!

//2.遍历,只加入下标
st.push(0);
for(int i=1;i<temperatures.size();i++) {
    //情况一;当前栈顶元素T[st.top()]>遍历的元素T[i]
    if(temperatures[st.top()]>temperatures[i]) {
        st.push(i);
        continue;
    }
    //情况二;当前栈顶元素T[st.top()]==遍历的元素T[i]
    if(temperatures[st.top()]==temperatures[i]){
        st.push(i);
        continue;
    }
    //情况三;当前栈顶元素T[st.top()]>遍历的元素T[i]---有删除才有结果更新
    if(temperatures[st.top()]<temperatures[i]){
        while(st.empty()==false&&temperatures[st.top()]<temperatures[i]){
            result[st.top()]=i-st.top();
            st.pop();
        }
        st.push(i);
        continue;
    }
}
for (int i=0;i<nums2.size();i++) {
    auto item=nums2[i];
    //情况1,2:代表栈空 + 栈不空且栈顶>item
    if(stack.empty()==true||stack.top()>item) stack.push(item);
    //情况3:代表栈不空且栈顶<item
    else{
        //关键点:栈顶右边第一个最大的元素是item
        unorderedMap.emplace(stack.top(),item);
        stack.pop();
        i--;
    }
}



//正确做法(单调栈加哈希)
class Solution1 {
public:
    vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {
        //1.先遍历nums2利用单调栈把结果直接保存到哈希中
        unordered_map<int,int> unorderedMap;
        stack<int> stack;
        for (int i=0;i<nums2.size();i++) {
            auto item=nums2[i];
            //情况1,2:代表栈空 + 栈不空且栈顶>item
            if(stack.empty()==true||stack.top()>item) stack.push(item);
            //情况3:代表栈不空且栈顶<item
            else{
                //关键点:栈顶右边第一个最大的元素是item
                unorderedMap.emplace(stack.top(),item);
                stack.pop();
                i--;
            }
        }
        while (stack.empty()== false){
            unorderedMap.emplace(stack.top(),-1);
            stack.pop();
        }


        //2.遍历nums1直接取结果
        vector<int> ans;
        for (auto item : nums1) {
            ans.push_back(unorderedMap[item]);
        }
        return ans;
    }
};
  • 2
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值