解题思路:
因为这两个int型的vector中不包含重复元素,所以考虑使用c++中的unordered_map,对num2进行遍历,找到每一个元素的下一个更大元素,这样的话,时间复杂度就是O(n)。
如果使用暴力的方法:对nums1中的每个int,在nums2中去找下一个更大元素,时间复杂度将会是O(num1.size() * nums2.size()),太过暴力。
使用一个unordered_map和一个stack进行处理的方法如下:
- 大循环遍历vector nums2;
- 当栈不为空且栈顶元素小于nums2当前元素时,说明找到了栈顶元素的下一个更大元素,将其记录在unordered_map中;
- 如果栈为空,或者栈顶元素仍大于当前的nums2元素,则将当前的nums2元素入栈。
循环下来之后,nums2中的一些元素可能会没有相应的下一个更大元素,留在了栈中,但我们不管这个了,不用再将其对应-1存进unordered_map中(为了省空间)。而且,最终结果可以直接在nums1中进行更新(省空间)。
class Solution {
public:
vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {
unordered_map<int, int> a;
stack<int> b;
//遍历nums2
for(auto x: nums2){
while(!b.empty() && b.top() < x){
a[b.top()] = x;
b.pop();
}
b.push(x);
}
//将结果直接在nums1中进行更新
for(int i = 0; i < nums1.size(); i++){
if(a.count(nums1[i])) //利用count函数来看看unordered_map中是否有这个key,有的话返回1,无 则返回0
nums1[i] = (a[nums1[i]]);
else nums1[i] = -1;
}
return nums1;
}
};