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

「已注销」

已于 2023-12-30 17:12:11 修改

阅读量529

点赞数 10

文章标签：算法

于 2023-12-22 17:33:45 首次发布

本文链接：https://blog.youkuaiyun.com/m0_72832574/article/details/135133532

版权

文章介绍了使用单调栈解决两个问题：在给定温度数组中找到每个温度后面的第一个较大元素的位置，以及在一个数组中找到另一个数组中每个元素后面的第一个较大元素。通过递增栈实现，时间复杂度O(n)，空间复杂度O(n)。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

739 每日温度

496 下一个更大元素 |

739 每日温度

求后面第一个比他大的元素的位置，单调栈需要递增

求后面第一个比他小的元素的位置，单调栈需要递减

本题栈头到栈底的顺序应该从小到大

class Solution {
public:
    vector<int> dailyTemperatures(vector<int>& temperatures) {
        stack<int> st;//求的是后面第一个比他大的元素的位置，所以单调栈需要递增
        vector<int> res(temperatures.size(),0);//使得递增
        st.push(0);//存放temperatures的下标
        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()]){//确保单调栈递增
                    res[st.top()] = i - st.top();
                    st.pop();
                }
                st.push(i);
            }
        }
        return res;
    }
};

时间复杂度O(n)

空间复杂度O(n)

496 下一个更大元素 |

求后面第一个比他大的元素的位置，单调栈需要递增

求后面第一个比他小的元素的位置，单调栈需要递减

本题栈头到栈底的顺序应该从小到大

class Solution {
public:
    vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {
        stack<int>st;
        vector<int>res(nums1.size(),-1);
        unordered_map<int,int>mp;
        for(int i = 0;i < nums1.size();i++){//key:下标元素 value:下标
            mp[nums1[i]] = i;//将nums1中的数据存放到mp中
        }
        st.push(0);//存放下标
        for(int i = 1;i < nums2.size();i++){
            if(nums2[i] <= nums2[st.top()]){//求后面第一个比他大的元素的位置，单调栈如果递增
                st.push(i);
            }else{//确保单调栈单调递增
                while(!st.empty() && nums2[i] > nums2[st.top()]){//找到自己的位置才停止
                    if(mp.count(nums2[st.top()]) > 0){//nums1中存在这个元素
                        int idx = mp[nums2[st.top()]];
                        res[idx] = nums2[i];
                    }
                    st.pop();
                }
                st.push(i);
            }
            
        }
        return res;
    }
};

时间复杂度O(m+n)//m是nums1的长度，n是nums2的长度

空间复杂度O(n)