2023.8.28
本题用暴力双层for循环解会超时,所以使用单调栈来解决,本质上是用空间换时间。维护一个单调递减栈,存储的是数组的下标。 代码如下:
class Solution {
public:
vector<int> dailyTemperatures(vector<int>& temperatures) {
vector<int> ans(temperatures.size(),0);
stack<int> stk;
for(int i=temperatures.size()-1; i>=0; i--)
{
while(!stk.empty() && temperatures[i]>=temperatures[stk.top()])
{
stk.pop();
}
if(!stk.empty()) ans[i] = stk.top()-i;
stk.push(i);
}
return ans;
}
};