LeetCode刷题 - #739. 每日温度

739. 每日温度

难度中等

给定一个整数数组 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]

关于单调栈

在栈的先进后出基础上额外增加一个特性: 从栈底(栈顶)到栈顶（栈底）的元素是严格递增或严格递减。

• 单调递减栈: 从栈底到栈顶递减

只有比栈顶元素小的元素才能直接进栈，否则需要把栈中比当前元素小的元素都出栈，再将当前元素入栈

// 单调递减栈
Deque<Integer> stack = new ArrayDeque<>();
for(int i = 0; i < nums.length; i++) {
    while(!stack.isEmpty() && stack.peek() < nums[i]) {
        stack.pop();
    }
    stack.push(nums[i]);
}

• 单调递增栈: 从栈底到栈顶递增

只有比栈顶元素大的元素才能直接进栈，否则需要把栈中比当前元素大的元素都出栈，再将当前元素入栈

// 单调递增栈
Deque<Integer> stack = new ArrayDeque<>();
for(int i = 0; i < nums.length; i++) {
    while(!stack.isEmpty() && stack.peek() > nums[i]) {
        stack.pop();
    }
    stack.push(nums[i]);
}

Next Greater Number问题

• 求数组中 右 侧第一个比它 大 的元素时，使用 单调递减栈 ，从左往右遍历，当前元素比栈顶元素大的时候，栈顶元素才会出栈。
• 求数组中 右 侧第一个比它 小 的元素时，使用 单调递增栈，从左往右遍历。
• 求数组中 左 侧第一个比它 大 的元素时，使用 单调递减栈，从右往左遍历。
• 求数组中 左 侧第一个比它 小 的元素时，使用 单调递增栈，从右往左遍历。

思路

① 找右侧下一个更大的数，从左往右遍历，使用单调递减栈。 ② 找到的下一个更大的数的下标与当前元素下标相减，返回该值

最终代码

class Solution {
    public int[] dailyTemperatures(int[] temperatures) {
        if(temperatures == null || temperatures.length == 0) {
            return null;
        }

        int[] result = new int[temperatures.length];
        Deque<Integer> stack = new ArrayDeque<>();

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