单调栈算法

Posted 2020-12-12 lcbwwy

利用单调栈，可以找到从左/右遍历第一个比它小/大的元素的位置.

假设有一个单调栈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;
}