题目:每日温度. - 力扣(LeetCode)
题解:单调栈
可以观察到,该类问题满足单调栈的下一个更大元素问题,单调栈内的数据是单调递减的。
class Solution {
public:
vector<int> dailyTemperatures(vector<int>& temperatures) {
vector<int>ans(temperatures.size(), 0);
stack<int>s;
for(int i = 0; i < temperatures.size(); i ++)
{
while(!s.empty() && temperatures[i] > temperatures[s.top()])
{
int t = s.top();
s.pop();
ans[t] = i - t;
}
s.push(i);
}
return ans;
}
};
思考: 单调栈可以解决什么样的问题?
单调栈内存储的数据一定是单调的,单调栈通过一次遍历数组并在过程中动态地维护栈,使得达成单调栈内的数据单调递增或者单调递减,因此他的时间复杂度一定是O(n)。
单调栈主要用于以下几类问题:
- 下一个更大元素问题:找到数组中每个元素右边第一个比它大的元素。
- 下一个更小元素问题:找到数组中每个元素右边第一个比它小的元素。
- 最近的更大元素问题:找到数组中每个元素左边第一个比它大的元素。
- 最近的更小元素问题:找到数组中每个元素左边第一个比它小的元素。