给你两个 没有重复元素 的数组 nums1 和 nums2 ,其中nums1 是 nums2 的子集。
请你找出 nums1 中每个元素在 nums2 中的下一个比其大的值。如果不存在,对应位置输出 -1 。
输入: nums1 = [4,1,2], nums2 = [1,3,4,2].
输出: [-1,3,-1]
利用单调栈(单调递减)把nums2中每个元素的目标值的出来,存入哈希表。
对nums1中的元素,通过哈希表查找答案。
易错点:最后不存在的几个值没有被存入哈希表。
class Solution {
public:
vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {
stack<int> temp;
map<int,int> mymap;
vector<int> ans;
for(int i=0;i<nums2.size();i++){//单调栈找最近一个大于它的数,存进哈希表
while(!temp.empty()&&nums2[i]>nums2[temp.top()]){//大于栈顶时保存答案
mymap[nums2[temp.top()]]=nums2[i];//此时nums2[i]就是下一个比它大的值
temp.pop();//找到后出栈
}
temp.push(i);//空栈/小于栈顶时入栈
}
for(auto i:nums1)
ans.push_back(mymap[i]==0?-1:mymap[i]);//前面没存进map默认为0 易错点!!!!!!!!!
return ans;
}
};
京公网安备 11010502036488号