LeetCode 739. 每日温度(单调栈)

739. 每日温度

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

例如,给定一个列表 temperatures = [73, 74, 75, 71, 69, 72, 76, 73],你的输出应该是 [1, 1, 4, 2, 1, 1, 0, 0]。

提示:气温 列表长度的范围是 [1, 30000]。每个气温的值的均为华氏度,都是在 [30, 100] 范围内的整数。

思路

看到左/右边第一个小于/大于当前元素的数,用单调栈求解。
要求边的,从左到右遍历;要求边的,从右到左遍历。
要求小于,用单调递增栈;要求大于,用单调递减栈
对于每个元素,要求的结果是该元素进入单调栈之前栈尾的元素

代码

class Solution {
    public int[] dailyTemperatures(int[] T) {
        int n = T.length, i = 0;
        int[] ret = new int[n];
        LinkedList<Node> stack = new LinkedList<>();
        for (i=n-1; i>=0; --i) {
            while (!stack.isEmpty() && stack.getLast().val < T[i]) {
                stack.removeLast();
            }
            if (!stack.isEmpty()) {
                ret[i] = stack.getLast().idx - i;
            }
            stack.add(new Node(i, T[i]));
        }
        return ret;
    }
}

class Node {
    int idx, val;

    public Node(int idx, int val) {
        this.idx = idx;
        this.val = val;
    }
}

你可能感兴趣的:(LeetCode)