hdu 1506(单调栈)

利用单调栈,可以在O(n)的复杂度下算出一个数组每个元素左/右第一个比它大/小的元素的位置

#include
#include
#include
#include
#include
#define ll long long
using namespace std;
const int N = 1e5+9;
stack  s;
int l[N],a[N],r[N];
ll h[N];
int main()
{
    freopen("in.txt","r",stdin);
    ios::sync_with_stdio(false);
    while(true){
       // memset(l,0,sizeof(l));
       // memset(r,0,sizeof(r));
        int n; cin>>n; if(n==0) break;
        for(int i = 1; i<=n; i++) cin>>h[i];
        while(s.size()) s.pop();
        for(int i = 1; i<=n; i++){
            while(s.size()&&h[s.top()]>=h[i]) s.pop();
            if(s.empty()) l[i] = 0;
            else  l[i] = s.top();
            s.push(i);
        }
        while(s.size()) s.pop();
        for(int i = n; i>0; i--){
            while(s.size()&&h[s.top()]>=h[i]) s.pop();
            if(s.empty()) r[i] = n+1;
            else r[i] = s.top();
            s.push(i);
        }
    //    for(int i = 1; i<=n; i++)
    //        cout<

 

你可能感兴趣的:(单调栈)