栈---每日温度

最新推荐文章于 2024-12-17 19:58:00 发布

原创最新推荐文章于 2024-12-17 19:58:00 发布 · 355 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#栈

算法专栏收录该内容

22 篇文章

订阅专栏

该博客探讨如何利用栈的数据结构解决寻找每日温度中需要等待多久才能升高的天数问题。示例和代码实现展示了如何类比寻找下一个更大元素的方法来解决此问题。题目规定气温列表长度在1到30000之间，气温值在30到100的整数范围内。

题目
根据每日气温列表，请重新生成一个列表，对应位置的输入是你需要再等待多久温度才会升高的天数。如果之后都不会升高，请输入 0 来代替。
示例

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

代码实现
思路：可类比找下一个更大元素。
提示：气温列表长度的范围是 [1, 30000]。每个气温的值的都是 [30, 100] 范围内的整数。

import java.util.Stack;

class Solution {
    public int[] dailyTemperatures(int[] temperatures) {
        //默认初始值为0
        int[] array = new int[temperatures.length];
        //维护一个始终栈顶元素最小的序列
        Stack<StackElement> stack = new Stack<>();
        for(int i=0;i<temperatures.length-1;i++){
            StackElement stackElement = new StackElement();
            stackElement.index = i;
            stackElement.val = temperatures[i];
            stack.push(stackElement);
            int top = stack.peek().val;
            int next = temperatures[i+1];
            //栈顶元素与next比较，若next大于栈顶元素，则找到了下一个更大的元素
            //直到next小于栈顶元素或栈为空
            while (!stack.empty()&&top<next){
                stackElement = stack.pop();
                array[stackElement.index]=i+1-stackElement.index;
                if(!stack.empty()){
                    top = stack.peek().val;
                }
            }
        }
        return array;
    }
}

/**
 * 自定义数据结构
 * index 记录元素在原数组的下标
 * val 记录元素的值
 */
class StackElement{
    int index;
    int val;
}