题目描述
根据每日 气温
列表,请重新生成一个列表,对应位置的输入是你需要再等待多久温度才会升高的天数。如果之后都不会升高,请输入 0
来代替。
例如,给定一个列表 temperatures = [73, 74, 75, 71, 69, 72, 76, 73]
,你的输出应该是 [1, 1, 4, 2, 1, 1, 0, 0]
。
提示:气温
列表长度的范围是 [1, 30000]
。每个气温的值的都是 [30, 100]
范围内的整数。
思路
利用递减栈,创建一个index栈,栈中存的是元素在T中的下标。
遍历T,若下标为栈顶元素的元素小于当前遍历到的元素,计算当前元素下标与栈顶元素(是个下标)的差值即为下标为栈顶元素的温度的未来升高天数,注意这里是循环判断,直到当前元素的值大于以栈顶元素为下标的值;当下标为栈顶元素的元素大于当前遍历到的元素时,将当前元素的下标入栈即可。T遍历完毕,栈中存留的下标对应的元素所对应的天数都为0(已初始化过)。
代码
class Solution {
public:
vector<int> dailyTemperatures(vector<int>& T) {
if(T.empty())
return T;
int n = T.size();
vector<int> res(n,0);
stack<int> index;
for(int i=0;i<n;i++){
while(!index.empty() && T[i]>T[index.top()]){
res[index.top()] = i-index.top();
index.pop();
}
index.push(i);
}
return res;
}
};