介绍
在一个数组中,对于每个数找到左边离他最近比他小的数和右边离他最近比他小的数
思想
我们可以使用一个栈结构,保持这个栈的元素自底向上是由小到大的。这就是一个单调栈结构。
(1)如果栈为空或者栈顶元素小于当前需要加入到栈中的元素,则将当前元素加入到栈中
(2)如果栈不为空且当前需要加入到栈中的元素小栈顶元素,此时需要执行出栈操作。即:当前需要加入到栈中的元素为栈顶元素右边离他最近的最小值,然后将栈顶弹出,此时栈顶的元素为弹出元素左边离他最近的最小值。然后继续将栈顶元素与当前需要加入到栈中的元素进行比较,直到符合情况(1)时,将当前元素入栈,否则一直弹出。
(3)当遍历完所有元素时,如果此时栈仍然不为空,则将栈中元素依次出栈,此时出栈的元素右边离他最近的最小值不存在,左边离他最近的最小值是他下面的元素。当栈中只存在一个元素时,此时这个元素左右两边离他最近的最小值都不存在
单调栈的应用
完整代码
#include <iostream>
#include <stack>
#include <utility>
using namespace std;
pair<int, int> loc;
pair<int, int> *maxLoc(int arr[], int size){
pair<int, int> *res = new pair<int, int>[size];
stack<int> S;
for(int i = 0; i < size; i++){
if(S.empty() || arr[S.top()] < arr[i]){
S.push(i);
}else{
while(!S.empty() && arr[S.top()] > arr[i]){
int temp = S.top();
S.pop();
int L = S.empty() ? -1 : S.top();
res[temp] = make_pair(L,i);
}
S.push(i);
}
}
while(!S.empty()){
int temp = S.top();
S.pop();
int L = S.empty() ? -1 : S.top();
res[temp] = make_pair(L, -1);
}
return res;
}
int main()
{
int n;
cin >>n;
int *arr = new int[n];
for(int i = 0; i < n; i++) cin >> arr[i];
pair<int, int> *res = maxLoc(arr, n);
for(int i = 0; i < n; i++)
cout << res[i].first << " " << res[i].second << endl;
return 0;
}