代码随想录训练营第五十八天|LeetCode 739、496

最新推荐文章于 2024-09-14 18:52:56 发布

gin7773

最新推荐文章于 2024-09-14 18:52:56 发布

阅读量136

点赞数

分类专栏： LeetCode刷题文章标签： leetcode 算法

本文链接：https://blog.csdn.net/weixin_43133169/article/details/129493096

版权

LeetCode刷题专栏收录该内容

44 篇文章 0 订阅

订阅专栏

文章介绍了如何使用栈解决LeetCode上的两道题目：739每日温度，通过栈找到每个温度需要等待多少天才能升到更高；496下一个更大元素I，利用栈和unordered_map找出数组中每个元素的下一个更大值。

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

LeetCode 739每日温度

题目链接：739.每日温度

class Solution {
public:
    vector<int> dailyTemperatures(vector<int>& temperatures) {
        vector<int> ans(temperatures.size(), 0);
        stack<int> st;//放置元素对应位置下标
        st.push(temperatures.size() - 1);
        for (int i = temperatures.size() - 2; i >= 0; --i) {
            while (!st.empty() && temperatures[i] >= temperatures[st.top()]) {
                st.pop();
            }
            if (st.empty()) ans[i] = 0;
            else ans[i] = st.top() - i;
            st.push(i);
        }
        return ans;
    }
};

LeetCode 496下一个更大元素I

题目链接：496.下一个更大元素I

class Solution {
public:
    vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {
        unordered_map<int, int> umap;//用来放nums2中的元素和其对应的下一个更大元素
        stack<int> st;
        for (int i = nums2.size() - 1; i >= 0; --i) {
            while (!st.empty() && nums2[i] >= st.top()) {
                st.pop();
            }
            if (st.empty()) umap[nums2[i]] = -1;
            else umap[nums2[i]] = st.top();
            st.push(nums2[i]);
        }

        vector<int> ans(nums1.size());
        for (int i = 0; i < nums1.size(); ++i) {
            unordered_map<int, int>::iterator it = umap.find(nums1[i]);
            ans[i] = it->second;
        }
        return ans;
    }
};