Leetcode 739. 每日温度

作为一个已经入行4年的程序猿，深感自己太多懒惰。因此希望自己可以每天保持学习，能够有知识的积累。

题目

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

• 提示：

1 <= temperatures.length <= 105
30 <= temperatures[i] <= 100

解析

思路，本题是要寻找多少天后气温会升高。这种查找最近的一个大于自己的数或者小于自己的数，特别适合用单调栈来实现。因为最终需要的是多少天后气温升高，因此栈中，存放数据下标即可
比如，我们使用实例2 的数据来进行解析
输入: temperatures = [30,40,50,60]
输出: [1,1,1,0]

• i = 0 时，单调栈为空，因此 0 进栈
  • stack = [0]
  • answer = [0, 0, 0 , 0]
• i = 1 时，40 > 30 因此stack 出栈，40入栈
  • stack = [1]
  • answer[1 - 0, 0, 0, 0]
• i = 2 时，50> 40 因此stack 出栈，50入栈
  • stack = [2]
  • answer[1, 2- 1, 0, 0]
• i = 3 时，60 > 50 因此stack 出栈，60入栈
  • stack = [3]
  • answer[1, 1, 3- 2, 0]

因此答案是

• answer[1, 1, 1, 0]

代码实现

class Solution {
    public int[] dailyTemperatures(int[] temperatures) {
        int[] answer = new int[temperatures.length];
        Stack<Integer> stack = new Stack<>();
        stack.push(0);
        for (int i = 1; i < temperatures.length; i++) {
            // 循环栈，直到比较数，小于或等于栈顶的值
            while (!stack.isEmpty() && temperatures[stack.peek()] < temperatures[i]) {
                int diff = i - stack.peek();
                answer[stack.pop()] = diff;
            }
            stack.push(i);
        }
        return answer;
        
    }
}