根据每日 气温 列表,请重新生成一个列表,对应位置的输出是需要再等待多久温度才会升高超过该日的天数。如果之后都不会升高,请在该位置用 0 来代替。
例如,给定一个列表 temperatures = [73, 74, 75, 71, 69, 72, 76, 73],你的输出应该是 [1, 1, 4, 2, 1, 1, 0, 0]。
提示:气温 列表长度的范围是 [1, 30000]。每个气温的值的均为华氏度,都是在 [30, 100] 范围内的整数。
单调栈
遍历整个数组,使用一个栈来作为保存数组元素下标,并且维持栈的单调递减性(从栈底到栈顶递减)。每当遇到一个新元素,如果此时栈为空或者新元素下标的元素小于等于栈顶元素下标的元素,则将新元素入栈,否则将栈顶元素弹出,并且此时新元素下标到栈顶元素下标的距离就是需要等待的天数,一直重复这个过程直到新元素能够入栈为止。
C++代码
class Solution {
public:
vector<int> dailyTemperatures(vector<int>& T) {
int n = T.size();
vector<int> res(n);
stack<int> s;
for (int i = 0; i < n; ++i) {
while (!s.empty() && T[i] > T[s.top()]) {
res[s.top()] = i - s.top();
s.pop();
}
s.push(i);
}
return res;
}
};
Python代码
class Solution:
def dailyTemperatures(self, T: List[int]) -> List[int]:
s = []
res = [0 for _ in range(len(T))]
for i in range(len(T)):
while len(s) != 0 and T[i] > T[s[-1]]:
res[s[-1]] = i - s[-1]
s.pop()
s.append(i)
return res