单调栈
介绍
单调栈是一种特殊的数据结构。专门用于解决一类问题叫做 Next Greater Number的问题。
性质:单调栈本身还是栈,只是每一次入栈,都会使得栈的排列有序。(其实也就是不满足有序的元素都不要)
通过适配 stack 而来。
Next Greater Number
问题描述
找到数组中每一个数之后最近的一个比该数大的数字(两种情况,返回索引或返回数值)
vector<int> next_greater_elem(const vector<int> & nums){
vector<int> res(nums.size()); //为了倒序访问,提前确定大小方便索引
stack<pair<int,int>> s; //单调栈 : pair<数字,索引>
for(int i=nums.size()-1;i>=0;i--){
while(!s.empty() && s.top().first <= nums[i]) //如果栈顶端的数无法满足有序性,就全弹出
s.pop();
res[i] = (s.empty() == true ) ? 0 : s.top().second-i; //这里返回的是距离,也可返回索引或数值
s.push(pair<int,int>(nums[i],i)); //插入pair<数字,索引>
}
return res;
}
时间复杂度:O(N)
空间复杂度:O(N)
Leetcode 相关题目
学习自labuladong