
经典单调栈问题。
感觉自己对这类问题还是不太熟练,想了很久思路,还想了很久是单调递增栈还是单调递减栈......
方法是维护一个单调递减栈。先将结果result初始化为0,如果温度一直递减,那么result就不用变化了。
遍历每日温度,如果温度比栈首温度高,那么往回遍历直到温度比栈首温度底,遍历的日期结果设为当前日期和遍历日期之差,遍历完的日期弹出。
cpp
class Solution {
public:
vector<int> dailyTemperatures(vector<int>& temperatures) {
vector<int> result(temperatures.size(),0);
stack<int> down;
for(int i=0;i<temperatures.size();i++){
while(!down.empty()&&temperatures[down.top()]<temperatures[i]){
result[down.top()]=i-down.top();
down.pop();
}
down.push(i);
}
return result;
}
};