力扣496—下一个更大元素I

思路:用哈希表

哈希表中存放nums2中的元素和对应的下标,创建数组vec用来存放每次查询的结果,作为最后的返回值。

遍历数组nums1,在哈希表中找nums1[i],因为nums1是nums2的子集,所以一定能找到。

un_map[nums1[i]]代表该元素在nums2中的下标,用j来记录该元素在nums2中的下一个元素下标,j=un_map[nums1[i]]+1;从这开始遍历数组nums2,如果找到nums2[j]>nums1[i]的值,则说明本次查询的答案为nums2[j],找到后break结束循环。

如果循环遍历完后没有找到,则本次查询结果为-1,需要vec.push_back(-1),所以我们需要定义一个标记值flag用来知道是否在nums2中找到了更大元素。

代码:

class Solution {
public:
    vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {
        unordered_map<int,int>un_map;
        vector<int>vec;
        for(int i=0;i<nums2.size();i++){
            un_map[nums2[i]]=i;
        }
        for(int i=0;i<nums1.size();i++){
            if(un_map.find(nums1[i])!=un_map.end()){
                int j=un_map[nums1[i]]+1;
                int flag=0;//用来记录是否在哈希表找到大于nums1[i]的值,找到flag=1,没找到为0
                for(;j<nums2.size();j++){//在nums2中遍历nums1[i]后面的元素找下一个更大元素
                    if(nums2[j]>nums1[i]){
                        flag=1;//在哈希表找到大于nums1[i]的值,修改flag=1,用作标记
                        vec.push_back(nums2[j]);
                        break;
                    }
                }
                if(flag==0)vec.push_back(-1);//标记未被修改了,说明没有找到下一个更大元素,本次查询的答案为-1
            }
        }
        return vec;
    }
};

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值