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

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

题目:
nums1 中数字 x 的 下一个更大元素 是指 x 在 nums2 中对应位置 右侧 的 第一个 比 x 大的元素。
给你两个 没有重复元素 的数组 nums1 和 nums2 ,下标从 0 开始计数,其中nums1 是 nums2 的子集。
对于每个 0 <= i < nums1.length ,找出满足 nums1[i] == nums2[j] 的下标 j ,并且在 nums2 确定 nums2[j] 的 下一个更大元素 。如果不存在下一个更大元素,那么本次查询的答案是 -1 。

返回一个长度为 nums1.length 的数组 ans 作为答案,满足 ans[i] 是如上所述的 下一个更大元素 。

解析:
此题跟上题基本一致,只需要判断nums2中每个元素的下一个更大元素即可,然后在对应的查找值即可。

  • 我一开始的思路是将此题视为跟上题一样的解法,求出nums2的result数组,然后在其中查找nums1的值,然后再形成数组即可。

  • 后来做题发现:在遍历过程当中,当找到了某值的下一个更大元素,此时其实就可以直接进行赋值给最终的对应于nums1的result数组了。因此可以在nums1中查找此值是否存在,然后找出此值在nums1中的下标,然后result数组进行赋值即可,此时发现需要查找操作,和记录nums1中元素的下标,因此使用了字典。

判断元素是否在数组中存在(查找操作),同时数组中的元素还对应一个要使用或返回的相应值,此时可以用字典来存储数组元素。

代码:

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

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值