问题描述:
思路分析:
不考虑暴力遍历的方法,太笨比了,失去了刷题的意义。
这个题目是在栈相关的练习下,所以考虑借助栈来解决。 1、先从头开始遍历数组,依次将温度入栈; 2、如果当前温度大于栈顶温度,将栈顶温度出栈,且两个温度的下标之差就是所求的需要等待的天数。然后继续比较当前温度跟新的栈顶温度的大小,重复之前的操作,直到该温度小于栈顶元素,则将其入栈; 3、如果当前温度小于栈顶温度,则继续将下一个元素入栈,直到该温度大于栈顶温度,跳转至步骤2; 4、遍历直到数组结束; 其实上述步骤就是为了构建一个递减栈。
代码实现:
class Solution {
public:
vector
<int> dailyTemperatures(vector
<int>& temperatures
) {
int n
= temperatures
.size();
vector
<int> res(n
, 0);
stack
<int> st
;
for (int i
= 0; i
< temperatures
.size(); ++i
) {
while (!st
.empty() && temperatures
[i
] > temperatures
[st
.top()]) {
auto t
= st
.top();
st
.pop();
res
[t
] = i
- t
;
}
st
.push(i
);
}
return res
;
}
};