【題目】*739. 每日溫度
根據每日 氣溫 列表,請重新生成一個列表,對應位置的輸出是需要再等待多久溫度纔會升高超過該日的天數。如果之後都不會升高,請在該位置用 0 來代替。
例如,給定一個列表 temperatures = [73, 74, 75, 71, 69, 72, 76, 73],你的輸出應該是 [1, 1, 4, 2, 1, 1, 0, 0]。
提示:氣溫 列表長度的範圍是 [1, 30000]。每個氣溫的值的均爲華氏度,都是在 [30, 100] 範圍內的整數。
【解題思路1】暴力法
class Solution {
public int[] dailyTemperatures(int[] T) {
int[] res = new int[T.length];
for (int i = 0; i < T.length; i++) {
int j = i + 1;
while (j < T.length && T[i] >= T[j]) {
j++;
}
res[i] = j < T.length && T[i] < T[j] ? j - i : 0;
}
return res;
}
}
優化的暴力法,來自於官方題解評論區
class Solution {
/** 暴力優化 —— 自底向上 */
public int[] dailyTemperatures(int[] T) {
int n = T.length;
int[] res = new int[n];
for (int i = n - 2; i >= 0; --i) {
// j += res[j] ==> 當T[i]<T[j]時,往後遍歷比T[j]溫度高的情況
for (int j = i + 1; j < n; j += res[j]) {
// 找到溫度高的一天
if (T[i] < T[j]) {
res[i] = j - i;
break;
}
// res[j] == 0說明第j天以後沒有比第j天更高的溫度了
else if (res[j] == 0) {
res[i] = 0;
break;
}
}
}
return res;
}
}
【解題思路2】單調棧
單調棧動畫
維護一個存儲下標的單調棧,從棧底到棧頂的下標對應的溫度列表中的溫度依次遞減。如果一個下標在單調棧裏,則表示尚未找到下一次溫度更高的下標。
class Solution {
public int[] dailyTemperatures(int[] T) {
int[] res = new int[T.length];
Deque<Integer> stack = new LinkedList<Integer>();
for(int i = 0; i < T.length; i++) {
while(!stack.isEmpty() && T[i] > T[stack.peek()]) {
res[stack.peek()] = i - stack.pop();
}
stack.push(i);
}
return res;
}
}