题目描述
请根据每日 气温 列表 temperatures ,请计算在每一天需要等几天才会有更高的温度。如果气温在这之后都不会升高,请在该位置用 0 来代替。
题解
执行用时:42 ms, 在所有 Java 提交中击败了33.34%的用户
内存消耗:47.7 MB, 在所有 Java 提交中击败了69.26%的用户
// ↓
// [55,38,53,81,61,93,97,32,43,78] len = 9
// [9]
// 0 0 0 0 0 0 0 0 0 0
// ↓
// [55,38,53,81,61,93,97,32,43,78] len = 9
// [9, 8]
// 0 0 0 0 0 0 0 0 1 0
// ↓
// [55,38,53,81,61,93,97,32,43,78] len = 9
// [9, 8, 7]
// 0 0 0 0 0 0 0 1 1 0
// ↓
// [55,38,53,81,61,93,97,32,43,78] len = 9
// [6]
// 0 0 0 0 0 0 0 1 1 0
// ↓
// [55,38,53,81,61,93,97,32,43,78] len = 9
// [6, 5]
// 0 0 0 0 0 1 0 1 1 0
// ↓
// [55,38,53,81,61,93,97,32,43,78] len = 9
// [6, 5, 4]
// 0 0 0 0 1 1 0 1 1 0
// ↓
// [55,38,53,81,61,93,97,32,43,78] len = 9
// [6, 5, 3]
// 0 0 0 2 1 1 0 1 1 0
// ↓
// [55,38,53,81,61,93,97,32,43,78] len = 9
// [6, 5, 3, 2]
// 0 0 1 2 1 1 0 1 1 0
// ↓
// [55,38,53,81,61,93,97,32,43,78] len = 9
// [6, 5, 3, 2, 1]
// 0 1 1 2 1 1 0 1 1 0
// ↓
// [55,38,53,81,61,93,97,32,43,78] len = 9
// [6, 5, 3, 0]
// 3 1 1 2 1 1 0 1 1 0
class Solution {
public int[] dailyTemperatures(int[] temperatures) {
int[] result = new int[temperatures.length];
Stack<Integer> stack = new Stack<>();
for (int i = temperatures.length - 1; i >= 0; i--) {
while (!stack.isEmpty() && temperatures[i] >= temperatures[stack.peek()]) { // stack非空,且遍历数大于等于栈顶数
stack.pop(); // 则说明栈中数字全部小于遍历树,把栈数字排空
}
// 如果栈顶不为空,说明栈顶数一定大于遍历树:要求的就是两索引的差
result[i] = stack.isEmpty() ? 0 : stack.peek() - i;
// 存储当前元素的索引
stack.push(i);
// System.out.println(stack.toString());
// printFunc(result);
}
return result;
}
public void printFunc(int[] arr) {
StringBuilder sb = new StringBuilder();
for (int i = 0; i < arr.length; i++) {
sb.append(arr[i] + " ");
}
System.out.println(sb.toString());
}
}