算法题[单调栈]

每日温度[单调栈]

• 今天先来普及一下什末是单调栈?

通常是一维数组，要寻找任一个元素的右边或者左边第一个比自己大或者小的元素的位置，此时我们就要想到可以用单调栈了。

题目:

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

题解

class Solution {
    public int[] dailyTemperatures(int[] temperatures) {
        //计算长度
        int lens = temperatures.length;
        //定义存放结果的数组
        int[] res = new int[lens];
        //定义单调栈
        Stack<Integer> stack = new Stack<>(); 

        //单调栈里放进去得是下标
       
        for(int i = 0; i < lens; ++i){
            //如果栈不为空，或者该元素大于栈顶元素
            while(!stack.isEmpty() && temperatures[i] > temperatures[stack.peek()]){
                //计算i和栈顶元素下标的差值，就是需要等待的天数
                res[stack.peek()] =  i - stack.peek();
                //栈顶弹出
                stack.pop();
            }
            //加入栈
            stack.push(i);
        }
        return res;
    }
}