Leetcode496
1.问题描述
2.解决方案(单调栈加哈希)
a.思路
1.这道题简直和上一题【Leetcode单调栈】739. 每日温度(有关于单调栈类型题的深入理解概括!!)一模一样啊,都是求数组中元素右边第一个比自己大的元素,这不是和上一题一摸一样嘛,上一题只不过是求所以元素,这一题求的是子集,反正我全求出来不就好了,子集那就从结果中找到子集就好了!
2.先遍历nums2利用单调栈把nums2中元素右边第一个比自己大的值求出来并放入map中
3.然后遍历nums1直接根据key取结果
4.也可以归纳为就是哈希表的建立问题,目前我们遇见过,先遍历一遍某个集合数组等,直接建立哈希表,这道题目是利用了单调栈的特性很符合题意,也是遍历一遍集合然后建立哈希表。
b.代码
1.本题的思路和【Leetcode单调栈】739. 每日温度(有关于单调栈类型题的深入理解概括!!)是一样的,但是代码的实现有所不同!
2.上一题遇到元素大于栈顶元素,是利用while一直出栈并记录结果,而本题是出一次栈,i–,使得下一次还是这个元素,这样怎么说更优雅一点好像,也还行
3.还有上一题是记录下标,本题是记录元素都是可以的!
//2.遍历,只加入下标
st.push(0);
for(int i=1;i<temperatures.size();i++) {
//情况一;当前栈顶元素T[st.top()]>遍历的元素T[i]
if(temperatures[st.top()]>temperatures[i]) {
st.push(i);
continue;
}
//情况二;当前栈顶元素T[st.top()]==遍历的元素T[i]
if(temperatures[st.top()]==temperatures[i]){
st.push(i);
continue;
}
//情况三;当前栈顶元素T[st.top()]>遍历的元素T[i]---有删除才有结果更新
if(temperatures[st.top()]<temperatures[i]){
while(st.empty()==false&&temperatures[st.top()]<temperatures[i]){
result[st.top()]=i-st.top();
st.pop();
}
st.push(i);
continue;
}
}
for (int i=0;i<nums2.size();i++) {
auto item=nums2[i];
//情况1,2:代表栈空 + 栈不空且栈顶>item
if(stack.empty()==true||stack.top()>item) stack.push(item);
//情况3:代表栈不空且栈顶<item
else{
//关键点:栈顶右边第一个最大的元素是item
unorderedMap.emplace(stack.top(),item);
stack.pop();
i--;
}
}
//正确做法(单调栈加哈希)
class Solution1 {
public:
vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {
//1.先遍历nums2利用单调栈把结果直接保存到哈希中
unordered_map<int,int> unorderedMap;
stack<int> stack;
for (int i=0;i<nums2.size();i++) {
auto item=nums2[i];
//情况1,2:代表栈空 + 栈不空且栈顶>item
if(stack.empty()==true||stack.top()>item) stack.push(item);
//情况3:代表栈不空且栈顶<item
else{
//关键点:栈顶右边第一个最大的元素是item
unorderedMap.emplace(stack.top(),item);
stack.pop();
i--;
}
}
while (stack.empty()== false){
unorderedMap.emplace(stack.top(),-1);
stack.pop();
}
//2.遍历nums1直接取结果
vector<int> ans;
for (auto item : nums1) {
ans.push_back(unorderedMap[item]);
}
return ans;
}
};