Leetcode 739. 每日温度

参考资料

代码随想录

思路

1. 因为是在一个一维数组，求任一个元素右边第一个比自己大的元素，可以考虑使用单调栈（从栈头到栈底是递增的）。

2. 因为结果是需要相对位置，所以栈里存放的是元素下标。

3. 考虑三种情况

        3.1 当遍历的元素小于栈顶元素，将遍历的元素压入栈中

        3.2 当遍历的元素等于栈顶元素，将遍历的元素压入栈中。（因为求的是大于的数）

        3.3 当遍历的元素大于栈顶元素，将遍历元素的下标和栈顶元素相减得到相对位置，将相对位置存入结果中。同时将将栈顶元素弹出，将当前遍历的元素压入栈中。

难点

1. 对于存放结果的向量，需要先初始化。因为后面是对result[ i ]进行操作，没有初始化，就是对不是result的元素操作。

实现代码

class Solution {
public:
    vector<int> dailyTemperatures(vector<int>& temperatures) {
        // 递增栈（从栈头到栈底递增）
        stack<int> st;

        // result必须初始化，后面result的操作是进行result[i]
        vector<int> result(temperatures.size(), 0);
        
        st.push(0);
        for (int i = 1; i < temperatures.size(); ++i) {
            // 情况一
            if (temperatures[i] < temperatures[st.top()]) {
                st.push(i);
            // 情况二
            } else if (temperatures[i] == temperatures[st.top()]) {
                st.push(i);
            // 情况三
            } else if ( temperatures[i] > temperatures[st.top()]) {
                while (!st.empty() &&temperatures[i] > temperatures[st.top()]) {
                    result[st.top()] = i - st.top();
                    st.pop(); 
                }
                st.push(i);
            }
        }
        return result;
    }
};