【Leetcode】739. 每日温度

题目描述

请根据每日 气温 列表 temperatures ，请计算在每一天需要等几天才会有更高的温度。如果气温在这之后都不会升高，请在该位置用 0 来代替。

题解

执行用时：42 ms, 在所有 Java 提交中击败了33.34%的用户

内存消耗：47.7 MB, 在所有 Java 提交中击败了69.26%的用户

//                              ↓
// [55,38,53,81,61,93,97,32,43,78] len = 9
// [9]
// 0 0 0 0 0 0 0 0 0 0 
//                          ↓
// [55,38,53,81,61,93,97,32,43,78] len = 9
// [9, 8]
// 0 0 0 0 0 0 0 0 1 0 
//                       ↓
// [55,38,53,81,61,93,97,32,43,78] len = 9
// [9, 8, 7]
// 0 0 0 0 0 0 0 1 1 0 
//                    ↓
// [55,38,53,81,61,93,97,32,43,78] len = 9
// [6]
// 0 0 0 0 0 0 0 1 1 0 
//                 ↓
// [55,38,53,81,61,93,97,32,43,78] len = 9
// [6, 5]
// 0 0 0 0 0 1 0 1 1 0 
//              ↓
// [55,38,53,81,61,93,97,32,43,78] len = 9
// [6, 5, 4]
// 0 0 0 0 1 1 0 1 1 0 
//            ↓
// [55,38,53,81,61,93,97,32,43,78] len = 9
// [6, 5, 3]
// 0 0 0 2 1 1 0 1 1 0 
//        ↓
// [55,38,53,81,61,93,97,32,43,78] len = 9
// [6, 5, 3, 2]
// 0 0 1 2 1 1 0 1 1 0 
//      ↓
// [55,38,53,81,61,93,97,32,43,78] len = 9
// [6, 5, 3, 2, 1]
// 0 1 1 2 1 1 0 1 1 0 
//  ↓
// [55,38,53,81,61,93,97,32,43,78] len = 9
// [6, 5, 3, 0]
// 3 1 1 2 1 1 0 1 1 0 

class Solution {
    public int[] dailyTemperatures(int[] temperatures) {
        int[] result = new int[temperatures.length];
        Stack<Integer> stack = new Stack<>();
        for (int i = temperatures.length - 1; i >= 0; i--) {
            while (!stack.isEmpty() && temperatures[i] >= temperatures[stack.peek()]) {  // stack非空，且遍历数大于等于栈顶数
                stack.pop();  // 则说明栈中数字全部小于遍历树，把栈数字排空
            }
            // 如果栈顶不为空，说明栈顶数一定大于遍历树：要求的就是两索引的差
            result[i] = stack.isEmpty() ? 0 : stack.peek() - i;
            // 存储当前元素的索引
            stack.push(i);
            
            // System.out.println(stack.toString());
            // printFunc(result);
        }
        return result;
    }
    
    public void printFunc(int[] arr) {
        StringBuilder sb = new StringBuilder();
        for (int i = 0; i < arr.length; i++) {
            sb.append(arr[i] + " ");
        }
        System.out.println(sb.toString());
    }
}