单调栈
单调栈的单调增和单调减都是从栈顶至栈底来看的。
#include <stack>
#include <vector>
#include <iostream>
using namespace std;
stack<int, vector<int>> get_monotoneStack(vector<int>& arr);
int main() {
vector<int> arr{10,2,9,4,7,3,5 };
stack<int, vector<int>> monotoneStack = get_monotoneStack(arr);
while (!monotoneStack.empty()) {
cout << monotoneStack.top() << endl;
monotoneStack.pop();
}
return 0;
}
// 递增
stack<int,vector<int>> get_monotoneStack(vector<int>& arr) {
stack<int,vector<int>> monotoneStack;
for (auto& a : arr) {
// 重点是这一个循环,每个元素是必须要push进去的。
// 如果栈里面的元素没有比它大的了,那么栈清空,它进去。
// 如果栈里面的元素有比它大的,那直接进去,该元素是离它最近的比它大的元素。
while (!monotoneStack.empty() && a >= monotoneStack.top())
monotoneStack.pop();
monotoneStack.push(a);
}
return monotoneStack;
}
题目
vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {
map<int,int> location={};
stack<int,vector<int>> monotoneStack={};
for(int i = nums2.size() - 1;i >= 0;i--){
while(!monotoneStack.empty() && nums2[i] >= monotoneStack.top()){
monotoneStack.pop();
}
if (monotoneStack.empty()) location[nums2[i]] = -1;
else location[nums2[i]] = monotoneStack.top();
monotoneStack.push(nums2[i]);
}
vector<int> res={};
for(auto &n:nums1)
res.push_back(location[n]);
return res;
}