LeetCode-739-每日温度-单调栈

题目描述:给定一个整数数组 temperatures ,表示每天的温度,返回一个数组 answer ,其中 answer[i] 是指对于第 i 天,下一个更高温度出现在几天后。如果气温在这之后都不会升高,请在该位置用 0 来代替。

题目链接: LeetCode-739-每日温度-单调栈

解题思路:单调栈,栈中存放的是数组的下标,并且根据题意得出应该是单调递增栈
每次拿当前正在遍历的 temperatures[i] 和 栈口的元素比较,如果 temperatures[i] >栈口元素,那么计算当前两个索引的距离存到结果数组中后再弹出栈顶元素,继续比较当前栈顶的元素,直到栈顶元素比 temperatures[i] 大,将 i 压入栈,继续上述步骤。

代码实现:

class Solution {
    public int[] dailyTemperatures(int[] temperatures) {
        int len = temperatures.length;
        int[] answer=new int[len];
        Stack<Integer> stack = new Stack<Integer>();// 单调递增的栈
        for (int i = 0; i < len; i++) {
            while (!stack.isEmpty() && temperatures[i] > temperatures[stack.peek()]){// 必须是大于,等于也不行
                answer[stack.peek()]=i-stack.peek();
                stack.pop();
            }
            // 直接入栈,入栈的是 下标
            stack.push(i);
        }
        return answer;
    }
}

你可能感兴趣的:(LeetCode,leetcode,算法)