剑指 Offer II 038. 每日温度

概要

使用栈，遍历数组，比较当前数字是否比栈顶元素大。大则出栈处理，直到栈空或者当前数字小于栈顶数字时，将当前数字入栈。

题目

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

链接：https://leetcode.cn/problems/iIQa4I

思路

其实和行星碰撞的很像。

我们用一个栈来保存数组中每个元素的下标，从做开始遍历数组。

1. 如果栈为空，则入栈。
2. 如果栈不为空，比较栈顶和当前元素的大小
   1. 如果当前元素大，则用当前元素下标和栈顶下标相减，得到栈顶元素“需要等待的天数”。之后再继续和栈里的元素比较大小。
   2. 如果当前元素小，则入栈。

这样始终让栈顶的元素是最小的，一旦有更大值出现，就把栈中比他小的值都处理掉。

解法：单调栈

代码

public int[] dailyTemperatures2(int[] temperatures) {
    int[] result = new int[temperatures.length];
    Deque<Integer> stack = new ArrayDeque<>();
    for (int i = 0; i < temperatures.length; i++) {
        if (stack.isEmpty()) {
            stack.push(i);
            continue;
        }
        int current = temperatures[i];
        while (!stack.isEmpty() && current > temperatures[stack.peek()]) {
            int index = stack.pop();
            result[index] = i - index;
        }
        stack.push(i);
    }
    // 处理观测不到高温的日子
    while (!stack.isEmpty()) {
        result[stack.pop()] = 0;
    }

    return result;
}

执行结果

这里注意，直接使用了Deque来代替Stack，因为Stack的效率实在太低了……