标签：int res st vector Leetcode 496 nums1 nums2 刷题

496. 下一个更大元素 I

题目链接

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/next-greater-element-i/
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

题目描述

题目分析

两个 没有重复元素 的数组 nums1 和 nums2，所以可以用map做映射。
先找nums2的“最大”，找到后再映射(如果nums1中存在)到nums1。
注意：res中找的不是索引，是值。

class Solution {
public:
    vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {
        int n1 = nums1.size();
        int n2 = nums2.size();

        vector<int> res(n1, -1);
        if(n1 == 0) return res;

        stack<int> st;
        st.push(0);
        
        unordered_map<int, int> umap;
        for(int i = 0; i < n1; i++){
            umap[nums1[i]] = i;
        }
        for(int i = 1; i < n2; i++){
            while(!st.empty() && nums2[i] > nums2[st.top()]){//
                if(umap.count(nums2[st.top()]) > 0){//nums2的值有在nums1中的
                    int index = umap[nums2[st.top()]];
                    res[index] = nums2[i];
                }
                st.pop();
            }
            st.push(i);
        }
        return res;
    }
};

标签：int,res,st,vector,Leetcode,496,nums1,nums2,刷题
来源： https://blog.csdn.net/qq_42771487/article/details/120280897

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。