2022-8-8 剑指offer-单调栈

剑指 Offer II 038. 每日温度

难度中等54收藏分享切换为英文接收动态反馈

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

class Solution {
    public int[] dailyTemperatures(int[] temperatures) {
        ArrayDeque<Integer> stack=new ArrayDeque<>();
        int n=temperatures.length;
        int[] ans=new int[n];
        for (int i=0;i<n;i++){
            if (stack.isEmpty()){
                stack.push(i);
            }else if (temperatures[i]<temperatures[stack.peek()]){
                stack.push(i);
            }else{
                int top=stack.peek();
                while (!stack.isEmpty()&&temperatures[i]>temperatures[top]){
                    ans[top]=i-top;
                    stack.pop();
                    if (!stack.isEmpty()) top=stack.peek();
                }
                stack.push(i);
            }
        }
        return ans;
    }
}

思路：单调递减栈，注意存储的是下标而不是温度本身。遇到小的就进栈，遇到大的就将上一个小的答案更新，出栈，直到空或者遇到更大的。