力扣第48天--- 第739题、第496题

最新推荐文章于 2024-10-05 12:29:44 发布

凑合看吧

最新推荐文章于 2024-10-05 12:29:44 发布

阅读量163

点赞数 1

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

本文链接：https://blog.csdn.net/u013441272/article/details/133128285

版权

本文介绍了如何使用单调栈解决LeetCode中的两道题目：第739题利用单调栈找到数组中每个元素右侧的第一个更大元素，第496题则涉及在一个有序数组中找到另一个数组中每个元素的下一个更大元素。作者详细解释了算法逻辑和代码实现。

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

# 力扣第48天— 第739题、第496题

一、第739题–每日温度

单调栈里放的是下标

适用场景：对于数组中某一元素，寻找右边/左边第一个大于或者小于这个元素的位置。

单调栈作用：存放之前遍历过的元素----存放下标。

要注意单调栈里的递增递减关系----栈顶到栈底的单调顺序，为后一个元素与前一个元素的关系一致。譬如，后一个比前一个大，那么栈顶到栈底就单调递增，栈顶是最小的。

不符合条件就进栈，符合条件就出栈。

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;
    }
};

二、第496题–下一个更大元素 I

逻辑上，有点绕，有些综合，整体还行。

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