739. 每日溫度
給定一個整數數組 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]
提示:
1 <= temperatures.length <= 105
30 <= temperatures[i] <= 100
題解
根據題意,我們要去尋找下一個更高溫度的位置
如果后面沒有更高的用 0 表示
不難發現,所有答案為 0 的數據合并到一起是遞減的
因為是 0 的數據其后面都沒有比它更大的
那么我們就可以去維護一個單調遞減的單調棧
期間維護彈出的數據都是答案不為 0 的,彈出的同時可以計算它們的答案
因為要計算數據之間的位置關系,單調棧內記錄數據的下標
單調遞減單調棧記為 arr
遍歷 temperatures 數組,將其加入到棧中并維護棧的單調性
因為棧要保持單調遞減,期間彈出的數據都是更小的
那么這些數據的答案就是它們的下標之差
記錄這些答案
一次遍歷結束后所有答案都求出了(棧中最后剩下的都是0)
代碼如下↓
class Solution {
public:vector<int> dailyTemperatures(vector<int>& temperatures) {int n=temperatures.size();vector<int> arr;vector<int> res(n);for(int i=0;i<n;i++){while(arr.empty()==false && temperatures[i]>temperatures[arr.back()]){res[arr.back()]=i-arr.back();arr.pop_back();}arr.push_back(i);}return res;}
};