Leetcode496. 下一个更大元素 I

meeiuliuus

于 2024-03-17 09:46:58 发布

阅读量327

点赞数 9

分类专栏： # leetcode --- easy 文章标签：算法

本文链接：https://blog.csdn.net/qq_52313711/article/details/136776606

版权

leetcode --- easy 专栏收录该内容

82 篇文章 0 订阅

订阅专栏

题目：

代码(首刷自解 2024年3月17日）：

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