题目描述:给定一个数组,求数组中每个元素在它后面第一个比它大的元素是多少。 如[2,1,2,4,3],返回数组 [4,2,4,-1,-1]。解释:第一个 2 后面比 2 大的数是 4; 1 后面比 1 大的数是 2;第二个 2 后面比 2 大的数是 4; 4 后面没有比 4 大的数,填 -1;3 后面没有比 3 大的数,填 -1。
知识:单调栈。
vector<int> nextGreaterElement(vector<int>& nums) {
vector<int> ans; // 存放答案
stack<int> s;
for (int i = nums.size() - 1; i >= 0; i--) { // 倒着往栈里放
while (!s.empty() && s.top() <= nums[i]) { // 判定数的大小
s.pop(); // 小的出栈
}
ans[i] = s.empty() ? -1 : s.top(); // 这个元素身后的第一个比它大的数
s.push(nums[i]); // 进队,再和前面的数比大小
}
return ans;
}
题目2:
解题思路:第一步:运用单调栈的知识,将nums2数组中各个元素的下一个比它更大的元素是多少求出,并保存到map<int,int >cnt中;第二步:遍历nums1中的元素,找出cnt中对应的数;
class Solution {
public:
vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {
int m=nums1.size();
int n=nums2.size();
vector<int>ans;
map<int,int>cnt;
stack<int>a;
for(int i=n-1;i>=0;i--)
{
while(!a.empty()&&a.top()<=nums2[i])
a.pop();
cnt[nums2[i]]=a.empty()?-1:a.top();//元素当作cnt的下标
a.push(nums2[i]);
}
for(int j=0;j<m;j++)
{
ans.push_back(cnt[nums1[j]]);
}
return ans;
}
};
小结:学会了单调栈的原理知识,