栈 - LeetCode 739.每日温度

给定一个整数数组 temperatures ，表示每天的温度，返回一个数组 answer ，其中 answer[i] 是指对于第 i天，下一个更高温度出现在几天后。如果气温在这之后都不会升高，请在该位置用 0 来代替。

输入: temperatures = [73,74,75,71,69,72,76,73]
输出: [1,1,4,2,1,1,0,0]

输入: temperatures = [30,40,50,60]
输出: [1,1,1,0]

解题方案: 单调栈+下标差

class Solution {
    public int[] dailyTemperatures(int[] temperatures) {
        int[] answer = new int[temperatures.length];
        // 注意stack里存的是元素的下标
        Deque<Integer> stack = new LinkedList<Integer>();
        for(int i=0; i<temperatures.length;i++) {
            int temperature = temperatures[i];
            // 注意这里是while循环
            while(!stack.isEmpty() && temperature > temperatures[stack.peek()]) {
                int prevIndex = stack.pop();
                answer[prevIndex] = i - prevIndex;
            }
            stack.push(i);
        }
        return answer;
    }
}

注意，answer初始化完，其实它元素默认都是0

image.png

这里方便观看，把0先都去掉了

image.png
image.png
image.png
image.png

所以，answer最后结果是：

image.png