LeetCode739. 每日温度
题目链接:739. 每日温度 - 力扣(LeetCode)
文章讲解:代码随想录
视频讲解:单调栈,你该了解的,这里都讲了!LeetCode:739.每日温度_哔哩哔哩_bilibili
思路:这道题最简单也最直接的思路是用双指针,也就是两层for循环,不过这样会超时。于是我们想到了单调栈的思路。单调栈求解是这样的:首先把第一个数的下标放进去,然后放第二个数,如果小于等于第一个数,那么直接push进去,如果大于,那么直接while循环把所有小于这个数的头pop出栈,并且保存两个下标之差进result数组。直到不大于栈顶的元素,就push进去。
代码:
class Solution {
public:
vector<int> dailyTemperatures(vector<int>& temperatures) {
stack<int>st;
vector<int>result(temperatures.size(),0);
st.push(0);
for(int i=1;i<temperatures.size();i++){
if(temperatures[i]<=temperatures[st.top()]){
st.push(i);
}
else{
while(!st.empty()&&temperatures[i]>temperatures[st.top()]){
result[st.top()]=i-st.top();
st.pop();
}
st.push(i);
}
}
return result;
}
};
LeetCode496. 下一个更大元素 I
题目链接:496. 下一个更大元素 I - 力扣(LeetCode)
文章讲解:代码随想录
视频讲解:单调栈,套上一个壳子就有点绕了| LeetCode:496.下一个更大元素_哔哩哔哩_bilibili
思路:和上一题思路有点像,但是很绕。首先我们要用unordered_map来存储nums1中的元素和下标的对应关系,然后使用单调栈遍历nums2,然后当遇到当前元素比栈顶元素大的情况时,我们去umap中找到这个数和这个数的下标,把result[这个数的下标]保存为nums[i]。
代码:
class Solution {
public:
vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {
stack<int>st;
vector<int>result(nums1.size(),-1);
unordered_map<int,int>umap;
if(nums1.empty()){
return result;
}
for(int i=0;i<nums1.size();i++){
umap[nums1[i]]=i;
}
st.push(0);
for(int i=1;i<nums2.size();i++){
if(nums2[i]<=nums2[st.top()]){
st.push(i);
}
else{
while(!st.empty()&&nums2[i]>nums2[st.top()]){
if(umap.find(nums2[st.top()])!=umap.end()){
int index=umap[nums2[st.top()]];
result[index]=nums2[i];
}
st.pop();
}
st.push(i);
}
}
return result;
}
};
LeetCode503. 下一个更大元素 II
题目链接:503. 下一个更大元素 II - 力扣(LeetCode)
文章讲解:代码随想录
视频讲解:单调栈,成环了可怎么办?LeetCode:503.下一个更大元素II_哔哩哔哩_bilibili
思路:和每日温度那一道题的代码很像,只不过需要把所有i替换成i%nums.size(),这样做相当于在原来那个的后面又接上了一段一样的数组,这样就可以起到循环的效果。
代码:
class Solution {
public:
vector<int> nextGreaterElements(vector<int>& nums) {
vector<int>result(nums.size(),-1);
if(nums.empty()){
return result;
}
stack<int>st;
st.push(0);
for(int i=1;i<nums.size()*2;i++){
if(nums[i%nums.size()]<=nums[st.top()]){
st.push(i%nums.size());
}
else{
while(!st.empty()&&nums[i%nums.size()]>nums[st.top()]){
result[st.top()]=nums[i%nums.size()];
st.pop();
}
st.push(i%nums.size());
}
}
return result;
}
};