利用单调栈,可以找到从左/右遍历第一个比它小/大的元素的位置.
假设有一个单调栈S和一个数组a[5];
有一个记录数组L[5],序号由1-5,表示a[i]向左遍历第一个比a[i]小的数的下标;
a[5]:3 5 7 4 6
当a[1]=3压入栈中,是空栈,L[1]=0;
当a[2]=5压入栈中,5>3,L[2]=1;
当a[3]=7压入栈中,7>5,L[3]=2;
当a[4]=4压入栈中,4<7,弹出7,4<5,弹出5,4>3,L[4]=1;
当a[5]=6压入栈中,6>4,L[3]=4;
L[5]:0 1 2 1 4
#include<bits/stdc++.h>
using namespace std;
int a[100];
int L[100];//存储向左遍历第一个比a[i]小的数的位置
stack <int>s;
int main( )
{
int n;
cin>>n;
for(int i=0;i<n;i++)
{
cin>>a[i];
}
for(int i=0;i<n;i++)
{
while(!s.empty( )&&a[s.top( )]>a[i])//单调递增栈
{
s.pop( );
}
if(s.empty( ))
{
L[i]=0;
}
else
{
L[i]=s.top( );
}
s.push(i);
}
for(int i=0;i<n;i++)
{
cout<<L[i]<<" ";
}
cout<<endl;
return 0;
}
网友评论