问题描述
给定两个 没有重复元素 的数组 nums1 和 nums2 ,其中nums1 是 nums2 的子集。找到 nums1 中每个元素在 nums2 中的下一个比其大的值。
nums1 中数字 x 的下一个更大元素是指 x 在 nums2 中对应位置的右边的第一个比 x 大的元素。如果不存在,对应位置输出 -1 [ 1 ] ^{[1]} [1]。
解题报告
求解每个元素的下一个更大元素,采用 单调递减栈
来存储数组中的元素,当前元素小于栈顶元素时,压栈;当前元素大于栈顶元素时,当前元素即为栈顶元素的下一个更大元素,出栈,记录栈顶元素的下一个更大元素,直到栈顶元素小于当前元素。
将每个元素的下一个更大元素存到哈希表中,之后可以在 O ( 1 ) O(1) O(1) 的时间复杂度内查询。
实现代码
class Solution {
public:
vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {
unordered_map<int, int>mp;
stack<int>stk;
for(int i=0;i<nums2.size();i++){
while(!stk.empty()&&nums2[stk.top()]<nums2[i]){
mp[nums2[stk.top()]]=nums2[i];
stk.pop();
}
stk.push(i);
}
for(int i=0;i<nums1.size();i++){
if(mp.find(nums1[i])!=mp.end())
nums1[i]=mp[nums1[i]];
else
nums1[i]=-1;
}
return nums1;
}
};