代码随想录算法训练营第五十八天| 739. 每日温度、496.下一个更大元素 I

最新推荐文章于 2024-10-17 11:39:06 发布

xux1aot1an

最新推荐文章于 2024-10-17 11:39:06 发布

阅读量93

点赞数

文章标签：算法 leetcode 职场和发展

本文链接：https://blog.csdn.net/xux1aot1an/article/details/130621181

版权

文章讲述了如何使用单调栈数据结构解决两个编程问题：739每日温度和496下一个更大元素I。在每日温度问题中，通过建立递增栈，找到每个温度的下一个更高温度的天数；在下一个更大元素问题中，利用类似方法找到数组中每个元素右侧的最大值。哈希表在此过程中用于快速查找元素的索引。

摘要由CSDN通过智能技术生成

739 每日温度

题目链接：739
思路：定义一个从栈顶到栈底递增的栈，记录数组下标。若当前元素小于等于栈顶，说明未找到更高温度，直接入栈即可保持栈单调。若当前元素大于栈顶，说明找到了比栈顶大的元素，两者的下标差即为栈顶元素对应的下标的结果，记录该下标差，并弹出栈顶，继续判断新的栈顶，直至当前元素于等于栈顶，然后入栈。
代码：

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

496 下一个更大元素 I

题目链接：496
思路：参考739. 每日温度，用nums2来建立单调栈，若找到nums2中元素的右侧最大元素，则用哈希表寻找nums1中相同元素的下标，并记录在结果数组中。
代码：

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