遍历nums,并将元素存入单调递减栈,每次出栈的时候保存进map里
思路
遍历nums2,维护一个递减栈
当得到一个更大的数的时候,将栈里小于它的数都放到哈希表当中
举例来说,如果nums2里是3,2,1,4,那么栈前三位都是3,2,1,当遍历到4的时候,发现4比1大,这时候,哈希表里要添加hashmap[1] = 4,hashmap[2] = 4,hashmap[3] = 4。含义是,对于1这个数字而言,右边第一个比它大的数字是4。后面几个同理。
遍历nums1,对每一项查找哈希表,找到第一个比它大的数,并返回一个列表作为答案。如果在哈希表中不存在则返回默认值-1。
参考链接:https://leetcode-cn.com/problems/two-sum/solution/python3-wu-xing-dai-ma-di-jian-zhan-ha-xi-biao-by-/
class Solution {
public:
vector<int> nextGreaterElement(vector<int>& findNums, vector<int>& nums) {
stack<int> s;
map<int,int> s_map;
vector<int> res;
for(int i=nums.size()-1;i>=0;--i)
{
while(!s.empty() && s.top()<=nums[i]) //栈里不存比当前值小的元素,保证栈内元素是有序的,栈顶元素最小,若存在比当前元素大的一定先出现在栈顶
s.pop();
s_map[nums[i]]=s.empty()?-1:s.top();
s.push(nums[i]);
}
for(int i=0;i<findNums.size();++i)
res.push_back(s_map[findNums[i]]);
return res;
}
};
转自
https://blog.csdn.net/liuchuo/article/details/54948152