思路:用哈希表
哈希表中存放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;
}
};