力扣496. 下一个更大元素 I

由于nums1数组是nums2数组的子集,所以可以直接在nums2s数组把每个数的下一个更大数用map来标记
寻找一个数的下一个更大数可以用单调栈,从数组的末尾反向遍历,维护一个单调递减的栈,(也可以从左往右遍历,维护一个单调递增的栈,从栈顶到底部单调递增)

class Solution {
public:
    vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {
            unordered_map<int,int> m;
            stack<int> st;
            for(int i = nums2.size()-1; i >=0; i--){
                int num = nums2[i];
                while(!st.empty()&&num >= st.top()){
                    st.pop();
                }
                if(st.empty())
                m[num] = -1;
                else m[num] = st.top();
                st.push(num);
            }
            vector<int> ans(nums1.size());
            for(int i = 0; i < nums1.size(); i++){
                ans[i] = m[nums1[i]];
            }
            return ans;
    }
};```
posted @   binbin成长日记  阅读(4)  评论(0编辑  收藏  举报
相关博文:
阅读排行:
· 地球OL攻略 —— 某应届生求职总结
· 周边上新:园子的第一款马克杯温暖上架
· Open-Sora 2.0 重磅开源!
· 提示词工程——AI应用必不可少的技术
· .NET周刊【3月第1期 2025-03-02】
点击右上角即可分享
微信分享提示