739. 每日温度(中等)
思路
-
我们可以维持一个单调递减的栈,表示每天的温度;为了方便计算天数差,这里存放位置(下标),而非温度本身。因为温度可以通过访问数组下标获取。
-
从左向右遍历数组,对于每个日期 p ,如果 p 的温度比栈顶存储位置 q 的温度高,则我们取出 q,并记录 q 需要等待的天数,即
q-p
;重复这一过程,直到 p 的温度小于等于栈顶元素位置的温度(或者空栈)时,将 p 插入栈顶,然后考虑下一天。在这个过程中,栈内元素永远保持单调递减。 -
最后如果栈内剩余一些日期,则说明它们之后都没有出现更暖和的日期,置为 0。
代码
class Solution {
public:
vector<int> dailyTemperatures(vector<int>& temperatures) {
int n = temperatures.size();
vector<int> ans(n);
stack<int> st;
for(int i=0; i<n; ++i){
while(!st.empty() && temperatures[i] > temperatures[st.top()]){
ans[st.top()] = i - st.top();
st.pop();
}
st.push(i);
}
return ans;
}
};
收获
- 想到了单调栈的做法,但是我只想到了把温度存入栈,没想到可以存入下标。