題目
根據每日 氣溫 列表,請重新生成一個列表,對應位置的輸出是需要再等待多久溫度纔會升高超過該日的天數。如果之後都不會升高,請在該位置用 0 來代替。
例如,給定一個列表 temperatures = [73, 74, 75, 71, 69, 72, 76, 73],你的輸出應該是 [1, 1, 4, 2, 1, 1, 0, 0]。
提示:氣溫 列表長度的範圍是 [1, 30000]。每個氣溫的值的均爲華氏度,都是在 [30, 100] 範圍內的整數。
思路
審題: 找出數組中的每個元素下一個比自己大的要相隔多遠
題目例子: [73, 74, 75, 71, 69, 72, 76, 73]
比73 大的,走1步就比自己大(74)
比74 大的,走1步就比自己大(75)
比75 大的,走4步就比自己大(76)
比71 大的,走2步就比自己大(72)
比69 大的,走1步就比自己大(72)
比72 大的,走1步就比自己大(76)
比76 大的,走0步,後面沒有比我大的了
比72 大的,走0步,後面沒有比我大的了
- 明白題目之後應該可以立馬想到一種暴力,兩個for即可。
- 單調棧
將每個位置的下標入棧,每次都對比下當前是不是比棧裏的大,找到比棧頂值大的數就出棧,下標相減就是兩者之間的距離。
注意這裏的棧我們用Deque哦。
Stack 由於歷史原因依舊可以使用,但是不推薦,棧的話官方推薦用Deque.雖然它也不是完美的。
好奇的寶寶自己動手查查吖。
解法
暴力
class Solution {
public int[] dailyTemperatures(int[] T) {
int[] res = new int[T.length];
for (int i = 0; i < T.length; i++) {
for (int j = i+1; j < T.length; j++) {
int x = T[j]-T[i];
if (x>0) {
res[i] = j-i;
break;
}
}
}
return res;
}
}
單調棧
class Solution {
public int[] dailyTemperatures(int[] T) {
if (T.length==0) return new int[0];
int[] res = new int[T.length];
Deque<Integer> stack = new ArrayDeque<>();
for (int i = 0; i < T.length; i++) {
int temp = T[i];
while (!stack.isEmpty() && temp > T[stack.peek()]) {
int preIndex = stack.pop();
res[preIndex] = i-preIndex;
}
stack.push(i);
}
return res;
}
}