自己思路:
想到用两个栈,一个维护元素、另一个维护下标。但是还是无法处理有重复元素的问题(用哈希表来存储的时候)。所以就看了答案的思路。
答案思路:
从前往后遍历,维护一个单调栈。栈存放数组的下标。
①栈为空 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;}
};
运行结果: