主要思路
方法一:暴力
维护一个数组 tf[],始终存储每个温度第一次出现的下标.
从后向前遍历,对每一个温度T[i],在tf中找T[i]+1到100的第一次出现的最小下标,减去 i 即是要等待的天数,这样可行的原因是我们是从后向前遍历的,tf[]存储的下标始终是在i之后的.
方法二:单调栈
维护一个栈,其中存储的是温度数组下标
从前往后遍历,当前温度高于栈顶温度时,说明栈顶温度的下一个更高的温度就是当前温度,栈顶元素出栈即可,他已经找到了,重复上述操作,直到栈为空或者栈顶温度大于等于当前温度;
如果当前温度没有高于栈顶温度,直接入栈即可.
//方法一:暴力
class Solution {
public:
vector<int> dailyTemperatures(vector<int>& T) {
int len = T.size();
vector<int> ans(len,0),tf(101,INT_MAX);
for(int i = len-1;i>=0;--i) {
int idx = INT_MAX;
for(int j = T[i]+1;j<=100;++j) {
idx = min(idx,tf[j]);
}
if(idx!=INT_MAX) {
ans[i] = idx-i;
}
tf[T[i]] = i;//更新当前温度第一次出现的位置
}
return ans;
}
};
//方法二:单调栈
class Solution {
public:
vector<int> dailyTemperatures(vector<int>& T) {
int len = T.size();
vector<int> ans(len,0);
stack<int> stk;
for(int i = 0;i<len;i++) {
while(!stk.empty()&&T[i]>T[stk.top()]) {
ans[stk.top()] = i - stk.top();
stk.pop();
}
stk.push(i);
}
return ans;
}
};