[LeetCode]剑指 Offer II 038. 每日温度

题目

请根据每日 气温 列表 temperatures ，重新生成一个列表，要求其对应位置的输出为：要想观测到更高的气温，至少需要等待的天数。如果气温在这之后都不会升高，请在该位置用 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]

方法

单调栈

class Solution {
    public int[] dailyTemperatures(int[] temperatures) {
        Stack<Integer> stack = new Stack<>();
        int[] ret = new int[temperatures.length];
        for (int i = 0; i < temperatures.length; i++) {
            while (!stack.empty() && temperatures[stack.peek()] < temperatures[i]) {
                int index = stack.pop();
                ret[index] = i - index;
            }
            stack.push(i);
        }
        return ret;
    }
}

单调栈内存放temperature的下标

只要栈空，或者当前下标温度小于栈顶下标温度，就入栈

一旦当前下标温度大于栈顶下标温度，就pop()并且在对应的答案数组下标更新当前下标与栈顶下标的差值

这个过程要重复进行到栈空或者栈顶温度大于当前温度

如果没有比某个下标温度更大的温度时，就不更新答案数组对应的下标值，因为默认是0，刚好满足题目的要求。