leetcode 64 Next Greater Element I C++

最新推荐文章于 2026-06-16 20:19:21 发布

原创最新推荐文章于 2026-06-16 20:19:21 发布 · 454 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#c++ #LeetCode #算法

收录于

C++ 同时被 2 个专栏收录

133 篇文章

订阅专栏

算法

46 篇文章

订阅专栏

本文介绍了一种使用栈实现的高效算法，该算法用于找出数组中每个元素右侧第一个比它大的元素。通过构建一个映射表，可以快速查找对应的结果。

遍历第二个数组，用一个栈来构建一个数字和next greater一一对应的map即可。

    vector<int> nextGreaterElement(vector<int>& findNums, vector<int>& nums) {
        stack<int> st;
        unordered_map<int,int> mp;
        vector<int> vec;
        for (int i : nums) {
            while(!st.empty()) {
                int num = st.top();
                if(i>num) {
                    mp[num] = i;
                    st.pop();
                } else {
                    break;
                }
            }
            st.push(i);
        }
        
        for (int i : findNums) {
            if (mp.find(i) == mp.end()) {
                vec.push_back(-1);
            } else {
                vec.push_back(mp[i]);
            }
        }
        return vec;
    }