leetcode 64 Next Greater Element I C++

最新推荐文章于 2022-04-28 15:15:31 发布

wcxdell

最新推荐文章于 2022-04-28 15:15:31 发布

阅读量372

点赞数

分类专栏：算法 C++ 文章标签： c++ LeetCode 算法

本文链接：https://blog.csdn.net/a2331046/article/details/70188967

版权

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

133 篇文章 0 订阅

订阅专栏

算法

46 篇文章 0 订阅

订阅专栏

遍历第二个数组，用一个栈来构建一个数字和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;
    }