leetcode739. 每日温度 单调栈

自己思路：

想到用两个栈，一个维护元素、另一个维护下标。但是还是无法处理有重复元素的问题（用哈希表来存储的时候）。所以就看了答案的思路。

答案思路：

从前往后遍历，维护一个单调栈。栈存放数组的下标。

①栈为空 or 当前下标元素 <= 栈顶元素，入栈；

②当前下标元素 > 栈顶元素，就出栈，并计算它们的下标之差，存入到这个出栈元素对应的数组里面。

代码：

class Solution {
public:
    vector<int> dailyTemperatures(vector<int>& temperatures) {
        int n = temperatures.size();
        vector<int> ans(n);	// 设置存放当前下标温度之后的几天可以遇到高温度
        stack<int> st;	// 单调栈，用来存放元素下标
        int i = 0;
        while( i != temperatures.size()) {
            int t = temperatures[i];  // 获取当前下标元素
            // 如果栈为空 or t<= 栈顶元素，就入栈，并把指针后移
            if(st.empty() || t <= temperatures[st.top()]){
                st.push(i);
                i++;
            }
            // 如果 t>栈顶元素，那么就是栈顶这个温度找到了比它高的温度，出栈栈顶元素并计算它们之间的间隔天数，存入到与栈顶元素相关的数组中。
            else if (t > temperatures[st.top()]) {
                int loc = st.top();
                st.pop();
                ans[loc] = i - loc;
            }
        }
        return ans;
    }
};

运行结果：