leetcode739. 每日温度 单调栈

自己思路:

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

答案思路:

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

①栈为空 or 当前下标元素 <= 栈顶元素,入栈;

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

代码:

class Solution {
public:
    vector dailyTemperatures(vector& temperatures) {
        int n = temperatures.size();
        vector ans(n);	// 设置存放当前下标温度之后的几天可以遇到高温度
        stack 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;
    }
};

运行结果:

leetcode739. 每日温度 单调栈_第1张图片

 

你可能感兴趣的:(LeetCode,java,数据结构,算法)