LCR 038. 每日温度(中等)(主站739)

https://leetcode.cn/problems/iIQa4I/
https://leetcode.cn/problems/daily-temperatures/
难度：☆☆☆☆

题目：

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

示例:

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

思路：单调栈

1. 最终结果是一个列表，创建ans列表，里面创建len(temperatures)个0，再新建一个栈
2. 用单调栈储存下标，之所以储存下标是因为要求相差天数，我们可理解为索引之差
3. 遍历temperatures，判断当前温度和栈顶下标温度的大小
4. 如果当前温度大于栈顶温度，即找到了比栈顶大更的温度，更新ans，ans[栈顶下标] = 当前下标 - 栈顶下标，并弹出栈顶
5. 这里注意栈顶元素被弹出后，当前温度还要和新栈顶继续比较，直到当前温度小于栈顶温度
6. 如果当前温度小于栈顶温度，则将当前温度下标入栈，从栈底到栈顶的下标温度依次递减
7. ans中没有更新的元素，就是没有找到更高温度，默认0。

代码：

Python
时间复杂度：O(n)，其中 n 是温度列表的长度。
空间复杂度：O(n)，其中 n 是温度列表的长度。

class Solution:
    def dailyTemperatures(self, temperatures: List[int]) -> List[int]:
        n = len(temperatures)
        ans = [0] * n
        stack = []

        for i in range(n):
            while stack and temperatures[i] > temperatures[stack[-1]]:
                ans[stack[-1]] = i - stack[-1]
                stack.pop()
            stack.append(i)
        return ans

Java
时间复杂度：O(n)，其中 n 是温度列表的长度。
空间复杂度：O(n)，其中 n 是温度列表的长度。

class Solution {
    public int[] dailyTemperatures(int[] temperatures) {
        int n = temperatures.length;
        int[] ans = new int[n];
        Stack<Integer> stack = new Stack<>();

        for (int i = 0; i < n; i++) {
            while (!stack.isEmpty() && temperatures[i] > temperatures[stack.peek()]) {
                ans[stack.peek()] = i - stack.peek();
                stack.pop();
            }
            stack.push(i);
        }
        return ans;
    }
}