leetcode菜狗入门 | 739. 每日温度

每日温度

题目描述

根据每日 气温 列表，请重新生成一个列表，对应位置的输入是你需要再等待多久温度才会升高超过该日的天数。如果之后都不会升高，请在该位置用 0 来代替。

例如，给定一个列表 temperatures = [73, 74, 75, 71, 69, 72, 76, 73]，你的输出应该是 [1, 1, 4, 2, 1, 1, 0, 0]。

提示：气温 列表长度的范围是 [1, 30000]。每个气温的值的均为华氏度，都是在 [30, 100] 范围内的整数。

解题思路

利用栈，栈中存的是温度和它对应的下标，如果当前温度比栈顶温度高，出栈，相差的天数就是 当前下表 - 栈顶下标

代码

class Solution {
public:
    vector<int> dailyTemperatures(vector<int>& T) {
       int n = T.size();
       vector<int> ans(n , 0);
       stack<vector<int>> temp;
       for(int i = 0; i < n; i++){
           if(temp.empty()){
                temp.push({T[i], i});
           }
            else{
                while(!temp.empty() && T[i] > temp.top()[0]){
                    ans[temp.top()[1]] = i - temp.top()[1];
                    temp.pop();
                }
                temp.push({T[i], i});
            }
       }
       return ans;
    }
};