代码随想录训练营第五十八天| (单调栈)● 739. 每日温度 ● 496.下一个更大元素 I

文章介绍了如何使用单调栈解决编程问题,如739.每日温度中的温度天数计算和496.下一个更大元素,通过空间优化减少遍历次数。关键数据结构如栈和映射map在解题中的运用被详细解释。
摘要由CSDN通过智能技术生成

 739. 每日温度 

今天正式开始单调栈,这是单调栈一篇扫盲题目,也是经典题。

大家可以读题,思考暴力的解法,然后在看单调栈的解法。 就能感受出单调栈的巧妙

代码随想录

这道题使用单调栈来实现,通过栈来存储已经遍历过的元素,用空间换时间,使用一次遍历即可完成。在栈中是递增来存储元素的,从栈顶到栈底的元素大小是递增排序的。每遍历到一个元素就要将元素存入栈中,当存入栈的元素大于当前栈顶的元素,就要将栈顶元素弹出,并计算温度天数的差值,重复执行直至当前元素小于栈顶元素。直至遍历完成。

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

 496.下一个更大元素 I 

本题和 739. 每日温度 看似差不多,其实 有加了点难度。

代码随想录

这道题比上道题复杂了一点,有两个数组,nums1是nums2的子集,所以需要一个map来对nums1的元素进行映射,将nums1中的元素和下标存入map中,在nums遍历的过程中判断该元素在nums1中是否存在,同时要根据map的key和value来找到nums2中的元素再nums1中的下标。

vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {
        //初始化
        stack<int> st;
        vector<int> result(nums1.size(), -1);
        if (nums1.size() == 0) return result;

        //使用map映射元素
        unordered_map<int, int> map;
        for (int i = 0; i < nums1.size(); i++) {
            map[nums1[i]] = i;
        }

        //遍历元素
        for (int i = 0; i < nums2.size(); i++) {
                while (!st.empty() && nums2[st.top()] < nums2[i]) {
                    if (map.find(nums2[st.top()]) != map.end()) {
                        int index = map[nums2[st.top()]]; // 根据map找到nums2[st.top()] 在 nums1中的下标
                        result[index] = nums2[i];
                    }
                    st.pop();
                }
                st.push(i);
            }
        return result;
    }

  • 8
    点赞
  • 9
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值