739.每日温度(线性表)
根据每日气温列表,请重新生成一个列表,对应位置的输出是需要再等待多久温度才会升高超过该日的天数。如果之后都不会升高,请在该位置用 0 来代替。
例如,给定一个列表 temperatures = [73, 74, 75, 71, 69, 72, 76, 73],你的输出应该是 [1, 1, 4, 2, 1, 1, 0, 0]。
提示:气温 列表长度的范围是 [1, 30000]。每个气温的值的均为华氏度,都是在 [30, 100] 范围内的整数。
【分析】
最开始的时候想着从前面开始遍历,遍历过之后的数放到队列里面去,后来新扫描到的数都跟队尾的元素做一个比较,如果比队尾元素大,就更新队尾元素所在位置的答案,然后pop掉队尾。然后比较下一个队尾元素,直到队尾元素大于新扫描的数。
class Solution {
public:
struct node
{
int id, data;
node(int i, int d):id(i),data(d){}
};
vector<int> dailyTemperatures(vector<int>& T) {
vector<int> ans;
ans.resize(T.size());
vector<node> tmpv;
for(int i=0; i<T.size(); i++)
{
if(tmpv.size()==0)
{
tmpv.push_back(node(i,T[i]));
}
else
{
while(tmpv.size() && (*tmpv.rbegin()).data<T[i])
{
ans[(*tmpv.rbegin()).id] = i - (*tmpv.rbegin()).id;
tmpv.pop_back();
}
tmpv.push_back(node(i,T[i]));
}
}
while(tmpv.size())
{
ans[(*tmpv.rbegin()).id] = 0;
tmpv.pop_back();
}
return ans;
}
};
然后我才发现,诶,这不就是单调栈!
把以上的tmpv改成一个栈,那么栈里只有递减元素。这就是单调栈!
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;
}
};