给定一个整数数组 temperatures
,表示每天的温度,返回一个数组 answer
,其中 answer[i]
是指对于第 i
天,下一个更高温度出现在几天后。如果气温在这之后都不会升高,请在该位置用 0
来代替。
示例 1:
输入: temperatures = [73,74,75,71,69,72,76,73]
输出: [1,1,4,2,1,1,0,0]
示例 2:
输入: temperatures = [30,40,50,60]
输出: [1,1,1,0]
示例 3:
输入: temperatures = [30,60,90]
输出: [1,1,0]
提示:
1 <= temperatures.length <= 105
30 <= temperatures[i] <= 100
题意
求数组a[i]的a[j](a[j]>a[i])相隔多少个位置 j-i,没有比他大的就输出0、
思路
因为我们要求的是比当前元素大位置,那么我们可以倒序来求一个单调栈,从后往前记录一个单调递减的栈,那我们我们每到下一个位置的元素,我们就可以很快的找到比他大的第一个元素。然后得到的答案就是(栈顶-i)。
代码
cpp
class Solution {
public:
vector<int> dailyTemperatures(vector<int>& a) {
int n = a.size();
stack<int> st;
vector<int> res;
for(int i=n-1;i>=0;i--){
while(st.size() && a[i]>=a[st.top()]) st.pop();
if(st.size()) res.push_back(st.top()-i);
else res.push_back(0);
st.push(i);
}
reverse(res.begin(),res.end());
return res;
}
};