力扣题目
解题思路
java代码
力扣题目:
给定一个整数数组 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]
解题思路:
算法原理:
利用单调栈来解决此问题。单调栈保证栈内元素对应的温度值呈单调递减的趋势。
思路:
- 遍历输入的温度数组。
- 对于每个温度,如果栈非空且当前温度大于栈顶元素对应的温度,那么说明栈顶元素找到了下一个更高的温度。计算二者的索引差值,即为等待的天数,并将结果存入答案数组,然后将栈顶元素出栈。
- 无论如何,都将当前温度的索引压入栈中。
代码分析:
- 在
dailyTemperatures
方法中,首先创建一个结果数组ans
用于存储每个温度需要等待的天数,初始化为 0。 - 然后创建一个栈
stack
用于存储温度的索引。 - 接着通过循环遍历温度数组:
- 当栈非空且当前温度大于栈顶索引对应的温度时,进行处理。
- 处理时,取出栈顶索引,计算当前索引与栈顶索引的差值,存入结果数组,然后将栈顶索引出栈。
- 最后将当前索引压入栈中。
时间复杂度:O(n),因为只需要对温度数组进行一次遍历。
空间复杂度:O(n),主要是栈和结果数组的空间占用
java代码:
package org.example.mouth8;
import java.util.Stack;
public class Leetcode739 {
public static void main(String[] args) {
int[] temperatures = {73, 74, 75, 71, 69, 72, 76, 73};
int[] ints = dailyTemperatures(temperatures);
for (int anInt : ints) {
System.out.println(anInt);
}
}
public static int[] dailyTemperatures(int[] temperatures) {
int n = temperatures.length;
int[] ans = new int[n];
Stack<Integer> stack = new Stack<>();
for (int i = 0; i < n; i++) {
while (!stack.isEmpty() && temperatures[i] > temperatures[stack.peek()]) {
int prevIndex = stack.pop();
ans[prevIndex] = i - prevIndex;
}
stack.push(i);
}
return ans;
}
}
更多详细内容同步到公众号,感谢大家的支持!
每天都会给刷算法的小伙伴推送明日一题,并且没有任何收费项