目录
问题描述:
给定一个整数数组 temperatures
,表示每天的温度,返回一个数组 answer
,其中 answer[i]
是指对于第 i
天,下一个更高温度出现在几天后。如果气温在这之后都不会升高,请在该位置用 0
来代替。
示例 1:
输入: temperatures = [73,74,75,71,69,72,76,73] 输出: [1,1,4,2,1,1,0,0]
示例 2:
输入: temperatures = [30,40,50,60] 输出: [1,1,1,0]
示例 3:
输入: temperatures = [30,60,90] 输出: [1,1,0]
实现代码与解析:
单调栈:
class Solution {
public:
vector<int> dailyTemperatures(vector<int>& temperatures)
{
vector<int> result(temperatures.size(), 0);
stack<int> stk;
for(int i = 0; i < temperatures.size(); i++)
{
while(!stk.empty() && temperatures[i] > temperatures[stk.top()])
{
result[stk.top()] = i - stk.top();
stk.pop();
}
stk.push(i);
}
return result;
}
};
原理思路:
首先介绍一下单调栈,单调栈就是用于这种寻找一个元素左边或则右边第一个大于或小于的元素位置的,遇到这种题就考虑吧使用单调栈。
然后说一下如何实现单调栈,其实非常简单,所谓单调栈,那就是维持栈的单调,拿此题举例,我们要找的是右边第一个大于的元素,首先我们遍历数组,我们就将小于栈顶元素的下标入栈(注意是下标,这里求的是下标的差值),若大于栈顶元素,就说明我们找到了栈顶元素的第一个大于其的结果,求出结果再将栈顶元素出栈,然后循环到不再出栈,就将遍历到的元素入栈,后进行下一次循环。
其实简单来说就是相互消除的过程,有点像之前一个符号相消的栈题,只不过这里的消除条件变成了大于才消除(顺便输出结果),而且可以一个可以消除多格元素,自身还要入栈被消除一次。或者可以把每个元素看成胖子,元素大小就为其肥胖程度,每次加入一个胖子,若其大于左侧的胖子就会把左侧的胖子吃掉(直到吃不了),这好像之前再知乎上看到的例子,感觉挺形象的,这样就好理解多了吧,大家也可以手动模拟一下,进一步理解到位。
还有一下注意事项,就比如初始化问题,这里我们有的元素可能没有结果,也就是遍历到最后都没找到其大于它的值,根据题意就赋值为零,我们先将结果数组都初始化为零,这样最后就不用再把栈里没出栈的全弄出来赋值了。还有就是代码
while(!stk.empty() && temperatures[i] > temperatures[stk.top()])
里是while判断而不是if判断,因为一个元素可能大于里面多个元素,需要多次判断,直到不再符合条件为止。