LeetCode739：每日温度

题目描述

给定一个整数数组 temperatures ，表示每天的温度，返回一个数组 answer ，其中 answer[i] 是指对于第 i 天，下一个更高温度出现在几天后。如果气温在这之后都不会升高，请在该位置用 0 来代替。

解题思想

使用单调栈

代码

/*
    通常是一维数组，要寻找任一个元素的右边或者左边第一个比自己大或者小的元素的位置，
    此时我们就要想到可以用单调栈了。时间复杂度为O(n)。

    当前元素如果比栈顶元素大：将栈内比当前元素小的弹出并计算记录，然后将该元素入栈
    当前元素 <= 栈顶顶元素：当前元素入栈
*/
class Solution {
public:
    vector<int> dailyTemperatures(vector<int>& temperatures) {
        vector<int> res(temperatures.size());
        
        stack<int> stk;

        //放入栈中的是下标
        stk.push(0);
        for (int i = 1; i < temperatures.size(); i++) {
            {
                while (!stk.empty() && temperatures[i] > temperatures[stk.top()]) {
                    res[stk.top()] = i - stk.top();
                    stk.pop();
                }
                stk.push(i);
            }
        }
        
        return res;
    }
};