LeetCode 算法496 下一个更大的元素
题目
给你两个 没有重复元素 的数组 nums1 和 nums2 ,其中nums1 是 nums2 的子集。
请你找出 nums1 中每个元素在 nums2 中的下一个比其大的值。
nums1 中数字 x 的下一个更大元素是指 x 在 nums2 中对应位置的右边的第一个比 x 大的元素。如果不存在,对应位置输出 -1 。
解法1 暴力求解
class Solution {
public:
vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {
vector<int> ans(nums1.size(), -1);
for(int i = 0; i < nums1.size(); ++i){
int j = 0;
while(j < nums2.size() && nums1[i] != nums2[j]){
j++;
}
while(j < nums2.size() && nums1[i] >= nums2[j]){
j++;
}
if(j == nums2.size())
continue;
else
ans[i] = nums2[j];
}
return ans;
}
};
解法2 单调栈
class Solution {
public:
vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {
int n = nums2.size();
map<int, int> Map1;
stack<int> s;
int index;
for(int i = 0; i < n; ++i){
while(!s.empty() && nums2[i] > nums2[s.top()]){
index = s.top();
s.pop();
Map1[nums2[index]] = nums2[i];
}
s.push(i);
}
while(!s.empty()){
Map1[nums2[s.top()]] = -1;
s.pop();
}
vector<int> ans;
for(int i = 0; i < nums1.size(); ++i){
ans.push_back(Map1[nums1[i]]);
}
return ans;
}
};