LeetCode-739-每日温度-单调栈

Arbicoral2023-09-24 21:46

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

解题思路：单调栈，栈中存放的是数组的下标，并且根据题意得出应该是单调递增栈

每次拿当前正在遍历的 temperatures $i$ 和栈口的元素比较，如果 temperatures $i$ >栈口元素，那么计算当前两个索引的距离存到结果数组中后再弹出栈顶元素，继续比较当前栈顶的元素，直到栈顶元素比 temperatures $i$ 大，将 i 压入栈，继续上述步骤。
代码实现：

java 复制代码

class Solution {
    public int[] dailyTemperatures(int[] temperatures) {
        int len = temperatures.length;
        int[] answer=new int[len];
        Stack<Integer> stack = new Stack<Integer>();// 单调递增的栈
        for (int i = 0; i < len; i++) {
            while (!stack.isEmpty() && temperatures[i] > temperatures[stack.peek()]){// 必须是大于，等于也不行
                answer[stack.peek()]=i-stack.peek();
                stack.pop();
            }
            // 直接入栈，入栈的是 下标
            stack.push(i);
        }
        return answer;
    }
}