给定一个整数数组 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]


这道题的关键在于用一个栈来存“未被解决的”下标。每遍历到一个新的温度,让它从栈顶开始解决,把能解决的都出栈(while 循环)。这其实会自然形成一个单调栈。

C++15 linesExpandCollapse
vector<int> dailyTemperatures(vector<int>& temperatures) {
    int n = temperatures.size();
    vector<int> ans(n, 0);
    stack<int> stk;
    
    for (int i = 0; i < n; i++) {
        while (!stk.empty() && temperatures[i] > temperatures[stk.top()]) {
            int day = stk.top();
            stk.pop();
            ans[day] = i - day;
        }
        stk.push(i);
    }   
    return ans;
}