https://leetcode-cn.com/problems/next-greater-element-i/
思路:题目说的花里胡哨……其实就是对于
n
u
m
s
2
nums2
nums2的每一个元素,求出位于其右侧且大于它的一个元素的值就可以了,搞一个单调栈就行了,维护一个单调递减栈即可在线性复杂度内求出。但是考虑到
n
u
m
s
1
nums1
nums1是乱序的,所以需要再搞一个哈希表。
class Solution {
public:
vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {
//维护一个单调非升栈
stack<int> s;
unordered_map<int,int> m;
int siz1=nums1.size(),siz2=nums2.size();
for(int i=0;i<siz2;i++){
while(!s.empty()&&nums2[i]>s.top())
m[s.top()]=nums2[i],s.pop();
s.push(nums2[i]);
}
while(!s.empty())
m[s.top()]=-1,s.pop();
vector<int> ans(siz1);
for(int i=0;i<siz1;i++)
ans[i]=m[nums1[i]];
return ans;
}
};