739. 每日温度

发布于 2024-06-11  6 次阅读


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

思路:用一个栈存储温度的递减序列,存储的是元素的下标,在push元素时,判断元素和栈顶下标代表元素的值,如果大于则弹出栈顶元素,连续弹出直到栈顶元素大于当前要push的元素,然后把此元素压入栈中,存储答案时的计算ans[stack.peek()] = count - stack.peek()

class Solution {
    public int[] dailyTemperatures(int[] temperatures) {
		int n = temperatures.length;
		int[] ans = new int[n];
		LinkedList<Integer> stack = new LinkedList<>();
		int count = 0;
		while (count < n){
			while (!stack.isEmpty() && temperatures[count] > temperatures[stack.peek()]){
				ans[stack.peek()] = count - stack.peek();
				stack.remove();
			}
			stack.push(count++);
		}
		return ans;
	}
}