给你两个 **没有重复元素** 的数组 `nums1` 和 `nums2` ,其中`nums1` 是 `nums2` 的子集。
请你找出 `nums1` 中每个元素在 `nums2` 中的下一个比其大的值。
`nums1` 中数字 `x` 的下一个更大元素是指 `x` 在 `nums2` 中对应位置的右边的第一个比 `x` 大的元素。如果不存在,对应位置输出 `-1` 。
示例 1:
输入: nums1 = [4,1,2], nums2 = [1,3,4,2].
输出: [-1,3,-1]
解释:
对于 num1 中的数字 4 ,你无法在第二个数组中找到下一个更大的数字,因此输出 -1 。
对于 num1 中的数字 1 ,第二个数组中数字1右边的下一个较大数字是 3 。
对于 num1 中的数字 2 ,第二个数组中没有下一个更大的数字,因此输出 -1 。
思路:用map来存比这个数大的数,然后从后往前遍历nums2,用stack来寻找比该值大的一个数,如果栈不为空且当前的数>=栈顶的数字,就进行pop(),不满足条件退出循环,然后判断该数是否为空,入过为空map中的值为-1,否者下一个比该值大的数就是栈顶,然后将该数入栈。
代码如下:
class Solution {
public:
vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {
stack<int>st;
map<int ,int>mp;
vector<int>ans;
for(int i = nums2.size()-1 ; i >= 0 ; i--){
while(!st.empty()&&nums2[i]>=st.top()){
st.pop();
}
if(st.empty()){
mp[nums2[i]] = -1;
}
else{
mp[nums2[i]] = st.top();
}
st.push(nums2[i]);
}
for(int i =0 ; i < nums1.size() ; i++){
ans.push_back(mp[nums1[i]]);
}
return ans;
}
};