LeetCode 单调栈 739. 每日温度

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;}
};