739. 每日温度
给定一个整数数组 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
题解
根据题意,我们要去寻找下一个更高温度的位置
如果后面没有更高的用 0 表示
不难发现,所有答案为 0 的数据合并到一起是递减的
因为是 0 的数据其后面都没有比它更大的
那么我们就可以去维护一个单调递减的单调栈
期间维护弹出的数据都是答案不为 0 的,弹出的同时可以计算它们的答案
因为要计算数据之间的位置关系,单调栈内记录数据的下标
单调递减单调栈记为 arr
遍历 temperatures 数组,将其加入到栈中并维护栈的单调性
因为栈要保持单调递减,期间弹出的数据都是更小的
那么这些数据的答案就是它们的下标之差
记录这些答案
一次遍历结束后所有答案都求出了(栈中最后剩下的都是0)
代码如下↓
class Solution {
public:vector<int> dailyTemperatures(vector<int>& temperatures) {int n=temperatures.size();vector<int> arr;vector<int> res(n);for(int i=0;i<n;i++){while(arr.empty()==false && temperatures[i]>temperatures[arr.back()]){res[arr.back()]=i-arr.back();arr.pop_back();}arr.push_back(i);}return res;}
};